quantize_per_tensor

class torch.quantize_per_tensor(input, scale, zero_point, dtype)

Преобразует тензор с плавающей точкой в квантованный тензор с заданными масштабом и точкой отсчёта.

Параметры:

• input (Tensor) – тензор с плавающей точкой или список тензоров для квантования
• scale (float or Tensor) – масштаб, применяемый в формуле квантования
• zero_point (int or Tensor) – смещение целочисленного значения, соответствующего нулю с плавающей точкой
• dtype (torch.dtype) – требуемый тип данных возвращаемого тензора. Должен быть одним из квантованных типов данных: torch.quint8, torch.qint8, torch.qint32

Возвращает:

Новый квантованный тензор или список квантованных тензоров.

Тип возвращаемого значения:

Tensor

Пример:

>>> torch.quantize_per_tensor(torch.tensor([-1.0, 0.0, 1.0, 2.0]), 0.1, 10, torch.quint8)
tensor([-1.,  0.,  1.,  2.], size=(4,), dtype=torch.quint8,
       quantization_scheme=torch.per_tensor_affine, scale=0.1, zero_point=10)
>>> torch.quantize_per_tensor(torch.tensor([-1.0, 0.0, 1.0, 2.0]), 0.1, 10, torch.quint8).int_repr()
tensor([ 0, 10, 20, 30], dtype=torch.uint8)
>>> torch.quantize_per_tensor([torch.tensor([-1.0, 0.0]), torch.tensor([-2.0, 2.0])],
>>> torch.tensor([0.1, 0.2]), torch.tensor([10, 20]), torch.quint8)
(tensor([-1.,  0.], size=(2,), dtype=torch.quint8,
    quantization_scheme=torch.per_tensor_affine, scale=0.1, zero_point=10),
    tensor([-2.,  2.], size=(2,), dtype=torch.quint8,
    quantization_scheme=torch.per_tensor_affine, scale=0.2, zero_point=20))
>>> torch.quantize_per_tensor(torch.tensor([-1.0, 0.0, 1.0, 2.0]), torch.tensor(0.1), torch.tensor(10), torch.quint8)
tensor([-1.,  0.,  1.,  2.], size=(4,), dtype=torch.quint8,
   quantization_scheme=torch.per_tensor_affine, scale=0.10, zero_point=10)