Функция torch.quantize_per_tensor PyTorch
Справочное руководство Pytorch torch
torch.quantize_per_tensorЭто функция в PyTorch, используемая для создания квантованного тензора с по-тензорным квантованием.
Определение функции
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())
Другие расширения