PerGroup

class torch.ao.quantization.observer.PerGroup(group_size)

表示量化中的每個通道組粒度。

此粒度型別為每個 <group_size> 元素的組計算不同的量化引數。

例如，如果輸入張量的形狀為 [8, 16]，組大小為 4，則輸入張量將被重塑為 [64, 4]，為每 4 個元素的組計算量化引數，總共得到 64 個量化引數。

變數

group_size (int) – 每個量化組的大小