torch.Tensor.q_per_channel_axis

Tensor.q_per_channel_axis() → int

对于经过线性（仿射）通道量化处理的张量，返回该通道量化所应用的维度索引。