You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.ops.AddN.rst 1.1 kB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. mindspore.ops.AddN
  2. ===================
  3. .. py:class:: mindspore.ops.AddN(*args, **kwargs)
  4. 按元素将所有输入的Tensor相加。
  5. 所有输入Tensor必须具有相同的shape。
  6. **输入:**
  7. - **x** (Union(tuple[Tensor], list[Tensor])) - 输入tuple或list由多个Tensor组成,其数据类型为Number或bool,用于相加。
  8. **输出:**
  9. Tensor,与 `x` 的每个Tensor具有相同的shape和数据类型。
  10. **异常:**
  11. - **TypeError** - `x` 既不是tuple,也不是list。
  12. **支持平台:**
  13. ``Ascend`` ``GPU`` ``CPU``
  14. **样例:**
  15. >>> class NetAddN(nn.Cell):
  16. ... def __init__(self):
  17. ... super(NetAddN, self).__init__()
  18. ... self.addN = ops.AddN()
  19. ...
  20. ... def construct(self, *z):
  21. ... return self.addN(z)
  22. ...
  23. >>> net = NetAddN()
  24. >>> x = Tensor(np.array([1, 2, 3]), mindspore.float32)
  25. >>> y = Tensor(np.array([4, 5, 6]), mindspore.float32)
  26. >>> output = net(x, y, x, y)
  27. >>> print(output)
  28. [10. 14. 18.]