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:

\[output = input * weight^{T} + bias\]

Warning

This is an experimental API that is subject to change or deletion.

Parameters
  • input (Tensor) – Input Tensor of shape \((*, in\_channels)\), where \(*\) means any number of additional dimensions.

  • weight (Tensor) – The weight applied to the input. The shape is \((out\_channels, in\_channels)\) or \((in\_channels)\).

  • bias (Tensor, optional) – Additive biases to the output. The shape is \((out\_channels)\) or \(()\). Defaults: None, the bias is 0.

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.]]