線性層

class torch.ao.nn.qat.Linear(in_features, out_features, bias=True, qconfig=None, device=None, dtype=None)

一個附帶用於權重的 FakeQuantize 模組的線性模組，用於量化感知訓練。

我們採用與 torch.nn.Linear 相同的介面，詳情請參閱 https://pytorch.com.tw/docs/stable/nn.html#torch.nn.Linear 中的文件。

類似於 torch.nn.Linear，FakeQuantize 模組初始化為預設值。

變數

weight (torch.Tensor) – 用於權重的 fake quant 模組

classmethod from_float(mod, use_precomputed_fake_quant=False)

從浮點模組或 qparams_dict 建立 qat 模組 引數：mod 一個浮點模組，由 torch.ao.quantization 工具生成或直接由使用者提供