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 - inputother 不是Tensor、number.Number。

  • TypeError - inputother 都不是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.]