Linear

class Linear(in_features, out_features, bias=True, compute_mode='default', **kwargs)

一个支持 QAT 模式下 Linear 的 QATModule 。可以被 Observer 和 FakeQuantize 应用。

参数

• in_features (int) – 各输入样本的大小。

• out_features (int) – 各输出样本的大小。

• bias (bool) – 若为 False ， 该层不添加偏置。默认：True

classmethod from_float_module(float_module)

返回从浮点型 Module 实例转换而来的 QATModule 实例。

上一页

QATModule

下一页

Elemwise