torch.nn.functional.linear

torch.nn.functional.linear(input, weight, bias=None) → Tensor

Применяет линейное преобразование к входным данным: $y = xA^T + b$.

Этот оператор поддерживает 2-D weight с разреженной структурой

Предупреждение

Поддержка разреженных данных находится на стадии бета-тестирования, и некоторые комбинации размеров, типов данных и устройств могут быть не поддерживаемы или не поддерживать автоматическое дифференцирование. Если вы обнаружили недостающую функциональность, пожалуйста, отправьте запрос на её добавление.

Этот оператор поддерживает TensorFloat32.

Форма:

• Вход: $(*, in\_features)$ где * означает любое количество дополнительных измерений, включая отсутствие дополнительных измерений
• Вес: $(out\_features, in\_features)$ или $(in\_features)$
• Смещение: $(out\_features)$ или $()$
• Вывод: $(*, out\_features)$ или $(*)$, в зависимости от формы весов