PerAxis

class torch.ao.quantization.observer.PerAxis(axis)

表示量化中的逐轴粒度。

这种粒度类型沿着张量的指定轴计算不同的量化参数。

例如，如果输入张量的形状是 [8, 16] 且 axis=0，则量化参数将针对张量的每一行计算。总共有 8 个量化参数。

变量

axis (int) – 执行归约（reduction）的轴。