mindspore.mint.nn.functional.linear
- mindspore.mint.nn.functional.linear(input, weight, bias=None)
Applies the dense connected operation to the input. The dense function is defined as:
Warning
This is an experimental API that is subject to change or deletion.
- Parameters
- Returns
Output whose shape is determined by the shape of the input and the weight.
- Raises
- Supported Platforms:
Ascend
Examples
>>> import numpy as np >>> import mindspore >>> from mindspore import Tensor, mint >>> input = Tensor([[-1., 1., 2.], [-3., -3., 1.]], mindspore.float32) >>> weight = Tensor([[-2., -2., -2.], [0., -1., 0.]], mindspore.float32) >>> bias = Tensor([0., 1.], mindspore.float32) >>> output = mint.nn.functional.linear(input, weight, bias) >>> print(output) [[-4. 0.] [10. 4.]]