Linear

class Linear(in_features, out_features, bias=True, compute_mode='default', **kwargs)[源代码]

一个支持 QAT 模式下 LinearQATModule 。可以被 ObserverFakeQuantize 应用。

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

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

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

classmethod from_float_module(float_module)[源代码]

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