Función torch.quantize_per_tensor de PyTorch

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

torch.quantize_per_tensorEs una función de PyTorch utilizada para crear tensores cuantizados con cuantificación por tensor (per-tensor).

Definición de la función

torch.quantize_per_tensor(input, scale, zero_point, dtype)

Ejemplo de uso

Ejemplo

import torch

# Crear tensor de entrada
input = torch.tensor([[-1.0, 0.0], [1.0, 2.0]])

# Cuantificar por tensor
# scale: factor de escala de cuantificación
# zero_point: punto cero
# dtype: tipo de dato de cuantificación
x = torch.quantize_per_tensor(input, scale=0.1, zero_point=10, dtype=torch.quint8)

print("Tensor cuantizado:")
print(x)
print("Valores originales descuantificados:")
print(x.dequantize())

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

Otras extensiones