QFunctional

class torch.ao.nn.quantized.QFunctional [исходный код]

Класс-обёртка для квантованных операций.

Экземпляр этого класса можно использовать вместо префикса torch.ops.quantized. Пример использования приведён ниже.

Примечание

Этот класс не предоставляет хук forward. Вместо этого необходимо использовать одну из нижележащих функций (например, add).

Примеры:

>>> q_add = QFunctional()
>>> a = torch.quantize_per_tensor(torch.tensor(3.0), 1.0, 0, torch.qint32)
>>> b = torch.quantize_per_tensor(torch.tensor(4.0), 1.0, 0, torch.qint32)
>>> q_add.add(a, b)  # Equivalent to ``torch.ops.quantized.add(a, b, 1.0, 0)``

Допустимые имена операций:

• add
• cat
• mul
• add_relu
• add_scalar
• mul_scalar