bias

Calibrator that returns the bias of all collected tensors.

Classes

BiasCalibrator

Bias calibrator, tracks the bias of all tensors collected.

class BiasCalibrator

Bases: _Calibrator

Bias calibrator, tracks the bias of all tensors collected.

__init__(method='mean', axis=None)

Initialize.

Parameters:

• method (str) –

• axis (int | Tuple[int, ...] | None) –

collect_calib_bias(x)

Compute bias of input tensor along axis.

Parameters:

x (Tensor) –

compute_calib_bias()

Return the bias of all tensors collected.