megengine.functional.nn.linear

linear(inp, weight, bias=None, compute_mode='default')[源代码]

对输入张量进行线性变换。

更多信息参见 Linear。

参数

  • inp (Tensor) – 形状为 (N, in_features) 的输入张量。

  • weight (Tensor) – 形状为 (out_features, in_features) 的权重张量。

  • bias (Optional[Tensor]) – 形状为 (out_features,) 的偏置张量。默认：None

返回类型

Tensor

上一页

megengine.functional.nn.local_response_norm

下一页

megengine.functional.nn.multi_head_attention