评价此页

PerRow#

class torch.ao.quantization.observer.PerRow[source]#

在量化中表示逐行粒度。

这是逐轴量化的一种特殊情况,并且是Float8矩阵乘法特有的,其中输入使用块大小为 (1, …, input.shape[-1]) 的量化。权重使用块大小为 (1, weight.shape[1]) 的量化。