BatchMatMulActivation¶

class BatchMatMulActivation(batch, in_features, out_features, bias=True, nonlinear_mode='identity', **kwargs)[source]¶

A QATModule BatchMatMulActivation with QAT support.

classmethod from_float_module(float_module)[source]¶: Return a QATModule instance converted from a float Module instance.

previous

ConvTranspose2d

next

QuantStub