torch.nn.functional.linear

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

对传入的数据应用线性变换：$y = xA^T + b$。

此操作支持具有稀疏布局的二维weight

警告

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

此操作符支持 TensorFloat32。

形状：

• 输入: $(*, in\_features)$ 其中 * 表示任意数量的附加维度，包括无

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

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

• 输出: $(*, out\_features)$ 或 $(*)$, 基于权重的形状