|
123456789101112131415161718192021222324252627282930313233343536373839404142 |
- mindspore.ops.AddN
- ===================
-
- .. py:class:: mindspore.ops.AddN()
-
- 逐元素将所有输入的Tensor相加。
-
- 所有输入Tensor必须具有相同的shape。
-
- **输入:**
-
- - **x** (Union(tuple[Tensor], list[Tensor])) - Tensor组成的tuble或list,类型为 `bool_ <https://www.mindspore.cn/docs/api/zh-CN/master/api_python/mindspore.html#mindspore.dtype>`_ 或 `number <https://www.mindspore.cn/docs/api/zh-CN/master/api_python/mindspore.html#mindspore.dtype>`_ 。
-
- **输出:**
-
- Tensor,与 `x` 的每个Tensor具有相同的shape和数据类型。
-
- **异常:**
-
- - **TypeError** - `x` 既不是tuple,也不是list。
- - **ValueError** - `x` 中存在shape不同的Tensor。
-
- **支持平台:**
-
- ``Ascend`` ``GPU`` ``CPU``
-
- **样例:**
-
- >>> 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.]
|