torch.Tensor.q_per_channel_axis

Tensor.q_per_channel_axis() → int

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