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)$ 或 $(*)$，取決於權重的形狀