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