Linear

class torch.ao.nn.quantized.dynamic.Linear(in_features, out_features, bias_=True, dtype=torch.qint8) [исходный код]

Модуль линейного преобразования с динамическим квантованием, принимающий и возвращающий тензоры с числами с плавающей точкой. Мы используем тот же интерфейс, что и torch.nn.Linear; документацию см. на странице https://pytorch.org/docs/stable/nn.html#torch.nn.Linear.

Как и в torch.nn.Linear, атрибуты случайным образом инициализируются при создании модуля и позднее перезаписываются.

Переменные:

• weight (Tensor) – необучаемые квантованные веса модуля формы $(\text{out\_features}, \text{in\_features})$.
• bias (Tensor) – необучаемое смещение модуля с числами с плавающей точкой формы $(\text{out\_features})$. Если bias — True, значения инициализируются нулём.

Примеры:

>>> m = nn.quantized.dynamic.Linear(20, 30)
>>> input = torch.randn(128, 20)
>>> output = m(input)
>>> print(output.size())
torch.Size([128, 30])

classmethod from_float(mod, use_precomputed_fake_quant=False) [исходный код]

Создаёт модуль с динамическим квантованием из модуля с числами с плавающей точкой или qparams_dict.

Параметры:

mod (Module) – модуль с числами с плавающей точкой, созданный с помощью утилит torch.ao.quantization или предоставленный пользователем.

classmethod from_reference(ref_qlinear) [исходный код]

Создаёт модуль с динамическим квантованием (fbgemm/qnnpack) из эталонного квантованного модуля :param ref_qlinear: эталонный квантованный модуль, созданный с помощью :type ref_qlinear: Module :param функций torch.ao.quantization или предоставленный пользователем: