mindspore.ops.AddN

查看源文件
class mindspore.ops.AddN[源代码]

逐元素将所有输入的Tensor相加。

更多参考详见 mindspore.ops.addn()

输入:
  • x (Union(tuple[Tensor], list[Tensor])) - Tensor组成的tuple或list,类型为bool型或数值型。

输出:

Tensor,与 x 的每个Tensor具有相同的shape和数据类型。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, nn, ops
>>> class NetAddN(nn.Cell):
...     def __init__(self):
...         super(NetAddN, self).__init__()
...         self.addN = ops.AddN()
...
...     def construct(self, *z):
...         return self.addN(z)
...
>>> net = NetAddN()
>>> x = Tensor(np.array([1, 2, 3]), mindspore.float32)
>>> y = Tensor(np.array([4, 5, 6]), mindspore.float32)
>>> output = net(x, y, x, y)
>>> print(output)
[10. 14. 18.]