mindspore/docs/api/api_python/ops/mindspore.ops.Add.rst

55 lines
2.2 KiB
ReStructuredText
Raw Normal View History

2021-12-04 13:55:42 +08:00
mindspore.ops.Add
=================
2021-12-22 11:46:09 +08:00
.. py:class:: mindspore.ops.Add()
2021-12-04 13:55:42 +08:00
2021-12-22 11:46:09 +08:00
两个输入Tensor逐元素相加。
2021-12-04 13:55:42 +08:00
.. math::
out_{i} = x_{i} + y_{i}
2021-12-22 11:46:09 +08:00
.. note::
2022-07-21 17:36:37 +08:00
- 输入 `x` 和 `y` 遵循 `隐式类型转换规则 <https://www.mindspore.cn/docs/note/zh-CN/r1.6/operator_list_implicit.html>`_ ,使数据类型保持一致。
2021-12-22 11:46:09 +08:00
- 输入必须是两个Tensor或一个Tensor和一个Scalar。
- 当输入是两个Tensor时它们的数据类型不能同时是bool并保证其shape可以广播。
- 当输入是一个Tensor和一个Scalar时Scalar只能是一个常数。
2021-12-04 13:55:42 +08:00
**输入:**
2022-07-21 17:36:37 +08:00
- **x** (Union[Tensor, number.Number, bool]) - 第一个输入是一个number.Number、bool值或数据类型为 `number <https://www.mindspore.cn/docs/api/zh-CN/r1.6/api_python/mindspore.html#mindspore.dtype>`_`bool_ <https://www.mindspore.cn/docs/api/zh-CN/r1.6/api_python/mindspore.html#mindspore.dtype>`_ 的Tensor。
2021-12-22 11:46:09 +08:00
- **y** (Union[Tensor, number.Number, bool]) - 第二个输入当第一个输入是Tensor时第二个输入应该是一个number.Number或bool值或数据类型为number或bool_的Tensor。当第一个输入是Scalar时第二个输入必须是数据类型为number或bool_的Tensor。
2021-12-04 13:55:42 +08:00
**输出:**
2021-12-22 11:46:09 +08:00
Tensorshape与输入 `x``y` 广播后的shape相同数据类型为两个输入中精度较高的类型。
2021-12-04 13:55:42 +08:00
**异常:**
2021-12-22 11:46:09 +08:00
- **TypeError** - `x``y` 不是Tensor、number.Number或bool。
2021-12-04 13:55:42 +08:00
**支持平台:**
2021-12-04 20:36:47 +08:00
2021-12-04 13:55:42 +08:00
``Ascend`` ``GPU`` ``CPU``
**样例:**
>>> # 用例1: x和y都是Tensor。
>>> add = ops.Add()
>>> x = Tensor(np.array([1, 2, 3]).astype(np.float32))
>>> y = Tensor(np.array([4, 5, 6]).astype(np.float32))
>>> output = add(x, y)
>>> print(output)
[5.7.9.]
>>> # 用例2: x是Scalar Tensory是Tensor。
>>> add = ops.Add()
>>> x = Tensor(1, mindspore.int32)
>>> y = Tensor(np.array([4, 5, 6]).astype(np.float32))
>>> output = add(x, y)
>>> print(output)
[5. 6. 7.]
>>> # x的数据类型为int32y的数据类型为float32。
>>> # 输出的数据类型为高精度float32。
>>> print(output.dtype)
Float32