torch.nn.functional.linear

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

对输入数据应用线性变换： $y = xA^T + b$.

此运算支持具有 稀疏布局 的 2D weight

警告

稀疏支持是一个测试版功能，某些布局/dtype/设备组合可能不受支持，或者可能没有自动微分支持。如果您发现缺少功能，请打开功能请求。

此运算符支持 TensorFloat32.

形状

• 输入： $(*, in\_features)$ 其中 * 表示任意数量的额外维度，包括没有维度

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

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

• 输出：$(*, out\_features)$ 或 $(*)$，取决于权重的形状。