@dskhudia: Thank you for the reply. The quantized functional works.
I want to quantize both the weights  and the activations and run inference in Pytorch with my custom class.
Is there a function that calculates the scale/zero_points automatically for that?