FloatFunctional¶
- class torch.ao.nn.quantized.FloatFunctional[source]¶
浮点运算的状态收集器类。
此类的实例可以代替
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)``
- 有效的操作名称
add
cat
mul
add_relu
add_scalar
mul_scalar