FusedMovingAvgObsFakeQuantize¶
- class torch.ao.quantization.fake_quantize.FusedMovingAvgObsFakeQuantize(observer=<class 'torch.ao.quantization.observer.MovingAverageMinMaxObserver'>, quant_min=0, quant_max=255, **observer_kwargs)[原始碼][原始碼]¶
定義一個融合模組來觀察張量。
融合模組,用於觀察輸入張量(計算最小值/最大值),計算 scale/zero_point 並對張量進行偽量化。此模組使用類似 MovingAverageMinMaxObserver 的計算方式處理輸入,以計算最小值/最大值,進而計算 scale/zero_point。observer 中的 qscheme 輸入用於區分對稱/仿射量化方案。
此模組的輸出由 x_out = (clamp(round(x/scale + zero_point), quant_min, quant_max)-zero_point)*scale 給出
與
FakeQuantize
類似,並接受與基底類別相同的屬性。