PyTorch torch.quantize_per_tensor 関数
Pytorch torch リファレンスマニュアルtorch.quantize_per_tensorPyTorch でテンソルごと（per-tensor）の量子化を行う量子化テンソルを作成するための関数です。
関数定義
torch.quantize_per_tensor(input, scale, zero_point, dtype)
使用例
実例
import torch
# 入力テンソルを作成
input = torch.tensor([[-1.0, 0.0], [1.0, 2.0]])
# テンソルごとに量子化
# scale: 量子化スケール係数
# zero_point: ゼロ点
# dtype: 量子化データ型
x = torch.quantize_per_tensor(input, scale=0.1, zero_point=10, dtype=torch.quint8)
print("量子化テンソル:")
print(x)
print("逆量子化された元の値:")
print(x.dequantize())
# 入力テンソルを作成
input = torch.tensor([[-1.0, 0.0], [1.0, 2.0]])
# テンソルごとに量子化
# scale: 量子化スケール係数
# zero_point: ゼロ点
# dtype: 量子化データ型
x = torch.quantize_per_tensor(input, scale=0.1, zero_point=10, dtype=torch.quint8)
print("量子化テンソル:")
print(x)
print("逆量子化された元の値:")
print(x.dequantize())
その他の拡張