评价此页

PerGroup#

class torch.ao.quantization.observer.PerGroup(group_size)[source]#

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

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

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

变量

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