if you only need this in quantization aware training, you’ll need to define your own fake quantize module(https://github.com/pytorch/pytorch/blob/v1.3.1/torch/quantization/fake_quantize.py 7) and fake quantize op
(https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/native/quantized/fake_quant_per_tensor_affine.cpp)