Linear
-
class
torch.nn.qat.Linear(in_features, out_features, bias=True, qconfig=None, device=None, dtype=None)[source] A linear module attached with FakeQuantize modules for weight, used for quantization aware training.
We adopt the same interface as torch.nn.Linear, please see https://pytorch.org/docs/stable/nn.html#torch.nn.Linear for documentation.
Similar to torch.nn.Linear, with FakeQuantize modules initialized to default.
- Variables
~Linear.weight (torch.Tensor) – fake quant module for weight
-
classmethod
from_float(mod)[source] Create a qat module from a float module or qparams_dict
Args: mod a float module, either produced by torch.ao.quantization utilities or directly from user
