BatchMatMulActivation¶

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

支持 QAT 的 QATModule BatchMatMulActivation.

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

上一页

ConvBnRelu2d

下一页

QuantStub