FloatFunctional

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

Класс-сборщик состояния для операций с числами с плавающей точкой.

Экземпляр этого класса можно использовать вместо префикса torch. для некоторых операций. См. пример использования ниже.

Примечание

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

Примеры:

>>> f_add = FloatFunctional()
>>> a = torch.tensor(3.0)
>>> b = torch.tensor(4.0)
>>> f_add.add(a, b)  # Equivalent to ``torch.add(a, b)``

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