torch.nn.functional.linear

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

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

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

警告

稀疏支持是 Beta 功能，某些布局/数据类型/设备组合可能不受支持，或可能没有 Autograd 支持。如果您发现缺少功能，请提交功能请求。

此运算符支持TensorFloat32。

形状

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

• Weight：$(out\_features, in\_features)$ 或 $(in\_features)$

• Bias：$(out\_features)$ 或 $()$

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