Функция torch.dequantize в PyTorch


Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.dequantizeЭто функция в PyTorch, используемая для деквантования квантованного тензора в тензор с плавающей точкой.

Определение функции

torch.dequantize(input)

Пример использования

Пример

import torch

# Создание входного тензора
input = torch.tensor([[-1.0, 0.0], [1.0, 2.0]])

# Квантованный тензор
quantized = torch.quantize_per_tensor(input, scale=0.1, zero_point=10, dtype=torch.quint8)

print("Квантованный тензор:")
print(quantized)

# Деквантование
x = torch.dequantize(quantized)

print("Деквантованный тензор с плавающей запятой:")
print(x)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения