torch.nn.functional.linear

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

將線性轉換應用於傳入的資料：$y = xA^T + b$。

此操作支援具有稀疏佈局的2-D weight

警告

稀疏支援是一個 Beta 功能，某些佈局/dtype/裝置組合可能不支援，或者可能不支援 autograd。 如果您發現缺少功能，請開啟一個功能請求。

此運算子支援 TensorFloat32。

形狀

• 輸入：$(*, in\_features)$，其中 * 表示任意數量的附加維度，包括無

• 權重：$(out\_features, in\_features)$ 或 $(in\_features)$

• 偏差：$(out\_features)$ 或 $()$

• 輸出：$(*, out\_features)$ 或 $(*)$，基於權重的形狀