torch.Tensor.q_per_channel_axis

Tensor.q_per_channel_axis() → int

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