Fonction torch.quantize_per_channel de PyTorch

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

torch.quantize_per_channelC'est une fonction de PyTorch utilisée pour créer un tenseur quantifié avec une quantification par canal.

Définition de fonction

torch.quantize_per_channel(input, scales, zero_points, axis, dtype)

Exemple d'utilisation

Exemple

import torch

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

# Quantifier par canal
# scales : facteur d'échelle pour chaque canal
# zero_points : point zéro pour chaque canal
# axis : spécifie la dimension selon laquelle quantifier
scales = torch.tensor([0.1, 0.2])
zero_points = torch.tensor([10, 10])
axis = 0

x = torch.quantize_per_channel(input, scales, zero_points, axis, 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