Elemwise

class Elemwise(method, **kwargs)[源代码]

相当于 elemwise 算子的 Module. 可调用 quantize_qat 替换成 QATModule 型的 Elemwise.

参数

method – 要逐元素应用的操作,支持以下字符串。它会对浮点数进行正常的逐元素操作。