Elemwise

class Elemwise(method, **kwargs)

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

参数

method – the elemwise method, support the following string. It will do the normal elemwise operator for float.

上一页

Sequential

下一页

Concat