torch.Tensor.q_per_channel_axis

Tensor.q_per_channel_axis() → int

Если тензор квантуется линейным (аффинным) способом по каналам, возвращает индекс размерности, по которой применяется квантование по каналам.