PerAxis

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

表示量化中的逐轴粒度。

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

例如，如果输入张量的形状为 [8, 16]，axis=0，则量化参数是为张量的每一行计算的。共得到 8 个量化参数。

变量

axis (int) – 执行约简的轴。