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)$ или $(*)$, в зависимости от формы веса