mindspore.ops.Sub

class mindspore.ops.Sub[源代码]

逐元素用第一个输入Tensor减去第二个输入Tensor。

更多参考详见 mindspore.ops.sub()

输入:
  • x (Union[Tensor, number.Number, bool]) - 第一个输入,是一个number.Number、bool值或数据类型为 numberbool_ 的Tensor。

  • y (Union[Tensor, number.Number, bool]) - 第二个输入,当第一个输入是Tensor时,第二个输入应该是一个number.Number或bool值,或数据类型为number或bool_的Tensor。当第一个输入是Scalar时,第二个输入必须是数据类型为number或bool_的Tensor。

输出:

Tensor,shape与两个输入广播后的shape相同,数据类型为两个输入中精度较高的类型。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([1, 2, 3]), mindspore.int32)
>>> y = Tensor(np.array([4, 5, 6]), mindspore.int32)
>>> sub = ops.Sub()
>>> output = sub(x, y)
>>> print(output)
[-3 -3 -3]