mindspore.ops.AccumulateNV2

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

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

有关更多详细信息,请参阅: mindspore.ops.accumulate_n()

输入:
  • x (Union(tuple[Tensor], list[Tensor])) - AccumulateNV2的输入,由多个Tensor组成的tuple或list,其数据类型为数值型,且每个元素的shape必须相等。

输出:

Tensor,数据类型和shape与输入 x 的每个条目相同。

支持平台:

Ascend GPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops, nn
>>> class NetAccumulateNV2(nn.Cell):
...     def __init__(self):
...         super(NetAccumulateNV2, self).__init__()
...         self.accumulateNV2 = ops.AccumulateNV2()
...
...     def construct(self, *z):
...         return self.accumulateNV2(z)
...
>>> net = NetAccumulateNV2()
>>> 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.]