Fonction torch.dequantize de PyTorch
Manuel de référence PyTorch torch
torch.dequantizeIl s'agit d'une fonction de PyTorch utilisée pour déquantifier un tenseur quantifié en un tenseur à virgule flottante.
Définition de fonction
torch.dequantize(input)
Exemple d'utilisation
Exemple
import torch
# Créer le tenseur d'entrée
input = torch.tensor([[-1.0, 0.0], [1.0, 2.0]])
# Tenseur quantifié
quantized = torch.quantize_per_tensor(input, scale=0.1, zero_point=10, dtype=torch.quint8)
print("Tenseur quantifié :")
print(quantized)
# Déquantifier
x = torch.dequantize(quantized)
print("Tenseur flottant déquantifié :")
print(x)
# Créer le tenseur d'entrée
input = torch.tensor([[-1.0, 0.0], [1.0, 2.0]])
# Tenseur quantifié
quantized = torch.quantize_per_tensor(input, scale=0.1, zero_point=10, dtype=torch.quint8)
print("Tenseur quantifié :")
print(quantized)
# Déquantifier
x = torch.dequantize(quantized)
print("Tenseur flottant déquantifié :")
print(x)
Autres extensions