PerGroup

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

表示量化中的每个通道组粒度。

此粒度类型为每个 <group_size> 元素的组计算不同的量化参数。

例如，如果输入张量的形状为 [8, 16]，组大小为 4，则输入张量将被重塑为 [64, 4]，为每 4 个元素的组计算量化参数，总共得到 64 个量化参数。

变量

group_size (int) – 每个量化组的大小