torch.Tensor.q_per_channel_axis

Tensor.q_per_channel_axis() → int

Учитывая тензор, квантованный с помощью линейного (аффинного) канала-по-каналу квантования, возвращает индекс измерения, по которому применяется квантование канала-по-каналу.