torch.Tensor.q_per_channel_axis

Tensor.q_per_channel_axis() → int

給定一個通過線性 (仿射) 逐通道量化的 Tensor，返回應用逐通道量化的維度的索引。