YouTip LogoYouTip

Pytorch Torch Fake_Quantize_Per_Tensor_Affine

# Create input tensor x = torch.randn(2,3,4,5) # Define scale factor and zero point scale = torch.tensor(1.5) zero_point = torch.tensor(0) # Perform tensor-level fake quantization y = torch.fake_quantize_per_tensor_affine(x, scale, zero_point, quant_min=0, quant_max=255) print("Quantized shape:", y.shape)

Image 4: Pytorch torch Reference Pytorch torch Reference

← Pytorch Torch FlattenPytorch Torch Eye β†’