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