Elemwise

class Elemwise(method, **kwargs)

A Module to do elemwise operator. Could be replaced with QATModule version Elemwise using quantize_qat.

Parameters:

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