PerAxis

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

表示量化中的逐軸粒度。

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

例如，如果輸入張量的形狀是 [8, 16] 且 axis=0，則量化引數將針對張量的每一行計算。總共有 8 個量化引數。

變數

axis (int) – 執行歸約（reduction）的軸。