mindspore.ops.dense

mindspore.ops.dense(input, weight, bias=None)[源代码]

对输入 input 应用全连接操作。全连接定义为：

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

警告

这是一个实验性API，后续可能修改或删除。

参数：

  • input (Tensor) - 输入Tensor，shape是 \((*, in\_channels)\)，其中 \(*\) 表示任意的附加维度。

  • weight (Tensor) - 输入Tensor的权重，shape是 \((out\_channels, in\_channels)\) 或 \((in\_channels)\)。

  • bias (Tensor，可选) - 添加在输出结果的偏差，shape是 \((out\_channels)\) 或 \(()\)。默认值：None ，偏差为0。

返回：

输出结果，shape由 input 和 weight 的shape决定。

异常：

  • TypeError - input 不是Tensor。

  • TypeError - weight 不是Tensor。

  • TypeError - bias 不是Tensor。

支持平台：

Ascend GPU CPU

样例：

>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input = mindspore.Tensor([[-1., 1., 2.], [-3., -3., 1.]], mindspore.float32)
>>> weight = mindspore.Tensor([[-2., -2., -2.], [0., -1., 0.]], mindspore.float32)
>>> bias = mindspore.Tensor([0., 1.], mindspore.float32)
>>> output = mindspore.ops.dense(input, weight, bias)
>>> print(output)
[[-4.  0.]
 [10.  4.]]