mindspore.ops.subtract
- mindspore.ops.subtract(input, other, *, alpha=1)[源代码]
对Tensor进行逐元素的减法。
\[output[i] = input[i] - alpha * other[i]\]- 参数:
input (Union[Tensor, number.Number]) - 参与减法的Tensor或者Number。
other (Union[Tensor, number.Number]) - 参与减法的Tensor或者Number。
- 关键字参数:
alpha (Number) - \(other\) 的乘数。默认值:
1
。
- 返回:
Tensor,shape与广播后的shape相同,数据类型为输入中精度较高的类型。
- 异常:
TypeError - input 或 other 不是Tensor、number.Number。
TypeError - input 和 other 都不是Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> input = Tensor(np.array([4, 5, 6]), mindspore.float32) >>> y = Tensor(np.array([1, 2, 3]), mindspore.float32) >>> z = ops.subtract(input, y, alpha=1) >>> print(z) [3. 3. 3.]