最大值

校准器，返回所有收集的张量的绝对最大值。

类

MaxCalibrator

最大校准器，全局跟踪最大值。

class MaxCalibrator

基础类：_Calibrator

最大校准器，全局跟踪最大值。

Parameters:

• calib_desc – 一个 MaxCalibDescriptor。

• num_bits – 一个整数。量化的位数。

• axis – 一个元组。参见 QuantizerAttributeConfig。

• unsigned – 一个布尔值。使用无符号量化。

Readonly Properties:

amaxs: amax的列表。Numpy数组被保存，因为它可能用于某些绘图。

__init__(num_bits=8, axis=None, unsigned=False, track_amax=False)

初始化。

property amaxs

返回迄今为止收集的amax列表。

collect(x)

跟踪所有张量的绝对最大值。

Parameters:

x – 一个张量

Raises:

RuntimeError – 如果amax的形状发生变化

compute_amax()

返回所有收集的张量的绝对最大值。

reset()

重置收集的绝对最大值。