mindspore.ops.accumulate_n

mindspore.ops.accumulate_n(x)[源代码]

逐元素计算列表中各个tensor的和。

mindspore.ops.accumulate_n() 与 mindspore.ops.addn() 类似，但accumulate_n不会等待其所有输入就绪后再求和，可节省内存。

参数：

x (Union(tuple[Tensor], list[Tensor])) - 输入tensors。

返回：

Tensor

支持平台：

Ascend GPU

样例：

>>> import mindspore
>>> x = mindspore.tensor([1, 2, 3]), mindspore.float32)
>>> y = mindspore.tensor([4, 5, 6]), mindspore.float32)
>>> output = mindspore.ops.accumulate_n([x, y, x, y])
>>> print(output)
[10. 14. 18.]