Elemwise

class Elemwise(method, **kwargs)

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

参数

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

上一页

Sequential

下一页

Concat