PyTorch torch.fake_quantize_per_tensor_affine Fonction

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

torch.fake_quantize_per_tensor_affineest une fonction de PyTorch utilisée pour effectuer une quantification factice au niveau des tenseurs sur les tenseurs.

Définition de fonction

torch.fake_quantize_per_tensor_affine(input, scale, zero_point, quant_min, quant_max)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur d'entrée
x = torch.randn(2, 3, 4, 5)

# Définir le facteur d'échelle et le point zéro
scale = torch.tensor(1.5)
zero_point = torch.tensor(0)

# Effectuer une quantification factice au niveau des tenseurs
y = torch.fake_quantize_per_tensor_affine(x, scale, zero_point, quant_min=0, quant_max=255)
print("Forme quantifiée :", y.shape)

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

Autres extensions