torch.Tensor.q_per_channel_axis

Tensor.q_per_channel_axis() → int

给定通过线性（仿射）按通道量化的 Tensor，返回应用按通道量化的维度索引。