PerAxis

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

表示量化中的逐軸粒度。

此粒度型別沿張量的指定軸計算不同的量化引數。

例如，如果輸入張量的形狀為 [8, 16]，axis=0，則量化引數是為張量的每一行計算的。共得到 8 個量化引數。

變數

axis (int) – 執行約簡的軸。