模型开发接口
推理部署接口
API 相关说明
Applies a linear transformation to the input tensor.
Refer to Linear for more information.
Linear
inp (Tensor) – input tensor with shape (N, in_features).
Tensor
weight (Tensor) – weight with shape (out_features, in_features).
bias (Optional[Tensor]) – bias with shape (out_features,). Default: None
Optional
previous
megengine.functional.nn.local_response_norm
next
megengine.functional.nn.multi_head_attention