megengine.functional.nn.linear

linear(inp, weight, bias=None, compute_mode='default')[source]

Applies a linear transformation to the input tensor.

Refer to Linear for more information.

Parameters

  • inp (Tensor) – input tensor with shape (N, in_features).

  • weight (Tensor) – weight with shape (out_features, in_features).

  • bias (Optional[Tensor]) – bias with shape (out_features,). Default: None

Return type

Tensor

previous

megengine.functional.nn.local_response_norm

next

megengine.functional.nn.multi_head_attention