PerGroup

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

表示量化中的逐通道分组粒度。

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

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

变量

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