Fonction torch.quantize_per_tensor de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.quantize_per_tensorC'est une fonction dans PyTorch utilisée pour créer un tenseur quantifié avec une quantification par tenseur.

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur d'entrée
input = torch.tensor([[-1.0, 0.0], [1.0, 2.0]])

# Quantifier par tenseur
# scale : facteur d'échelle de quantification
# zero_point : point zéro
# dtype : type de données de quantification
x = torch.quantize_per_tensor(input, scale=0.1, zero_point=10, dtype=torch.quint8)

print("Tenseur quantifié :")
print(x)
print("Valeurs originales déquantifiées :")
print(x.dequantize())

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions