Функция PyTorch torch.fake_quantize_per_tensor_affine
Справочное руководство PyTorch torch
torch.fake_quantize_per_tensor_affineявляется функцией в PyTorch, используемой для выполнения фиктивного квантования тензоров на тензорном уровне.
Определение функции
torch.fake_quantize_per_tensor_affine(input, scale, zero_point, quant_min, quant_max)
Пример использования
Пример
import torch
# Создать входной тензор
x = torch.randn(2, 3, 4, 5)
# Задать коэффициент масштабирования и нулевую точку
scale = torch.tensor(1.5)
zero_point = torch.tensor(0)
# Выполнить фиктивное квантование на уровне тензора
y = torch.fake_quantize_per_tensor_affine(x, scale, zero_point, quant_min=0, quant_max=255)
print("Квантованная форма:", y.shape)
# Создать входной тензор
x = torch.randn(2, 3, 4, 5)
# Задать коэффициент масштабирования и нулевую точку
scale = torch.tensor(1.5)
zero_point = torch.tensor(0)
# Выполнить фиктивное квантование на уровне тензора
y = torch.fake_quantize_per_tensor_affine(x, scale, zero_point, quant_min=0, quant_max=255)
print("Квантованная форма:", y.shape)
Другие расширения