Función torch.dequantize de PyTorch
Manual de referencia de PyTorch torch
torch.dequantizeEs una función de PyTorch utilizada para dequantizar tensores cuantizados a tensores de punto flotante.
Definición de la función
torch.dequantize(input)
Ejemplo de uso
Ejemplo
import torch
# Crear el tensor de entrada
input = torch.tensor([[-1.0, 0.0], [1.0, 2.0]])
# Cuantizar el tensor
quantized = torch.quantize_per_tensor(input, scale=0.1, zero_point=10, dtype=torch.quint8)
print("Tensor cuantizado:")
print(quantized)
# Dequantizar
x = torch.dequantize(quantized)
print("Tensor de punto flotante después de la dequantización:")
print(x)
# Crear el tensor de entrada
input = torch.tensor([[-1.0, 0.0], [1.0, 2.0]])
# Cuantizar el tensor
quantized = torch.quantize_per_tensor(input, scale=0.1, zero_point=10, dtype=torch.quint8)
print("Tensor cuantizado:")
print(quantized)
# Dequantizar
x = torch.dequantize(quantized)
print("Tensor de punto flotante después de la dequantización:")
print(x)
Otras extensiones