Función torch.fake_quantize_per_tensor_affine de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.fake_quantize_per_tensor_affineEs una función de PyTorch utilizada para realizar cuantización falsa (Fake Quantization) a nivel de tensor en tensores.

Definición de la función

torch.fake_quantize_per_tensor_affine(input, scale, zero_point, quant_min, quant_max)

Ejemplo de uso

Ejemplo

import torch

# Crear el tensor de entrada
x = torch.randn(2, 3, 4, 5)

# Definir el factor de escala y el punto cero
scale = torch.tensor(1.5)
zero_point = torch.tensor(0)

# Realizar cuantización falsa a nivel de tensor
y = torch.fake_quantize_per_tensor_affine(x, scale, zero_point, quant_min=0, quant_max=255)
print("Forma después de la cuantización:", y.shape)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones