PerAxis

class torch.ao.quantization.observer.PerAxis(axis) [исходный код]

Представляет гранулярность поканальной квантизации.

Этот тип гранулярности вычисляет различные параметры квантизации вдоль указанной оси тензора.

Например, если форма входного тензора — [8, 16], а axis=0, параметры квантизации вычисляются для каждой строки тензора. В итоге получается 8 параметров квантизации.

Переменные:

axis (int) – Ось, вдоль которой выполняется редукция.