Linear

class torch.ao.nn.qat.Linear(in_features, out_features, bias=True, qconfig=None, device=None, dtype=None) [исходный код]

Линейный модуль с модулями FakeQuantize для весов, предназначенный для обучения с учетом квантования.

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

Аналогично torch.nn.Linear, модули FakeQuantize инициализированы значениями по умолчанию.

Переменные:

weight (torch.Tensor) – модуль псевдоквантования для весов

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

Создает модуль QAT из модуля с плавающей точкой или qparams_dict. Аргументы: mod модуль с плавающей точкой, созданный с помощью утилит torch.ao.quantization или непосредственно пользователем