A Manual Implementation of Quantization in PyTorch - Single Layer