mindspore.ops.Sub
- class mindspore.ops.Sub[源代码]
逐元素用第一个输入Tensor减去第二个输入Tensor。
更多参考详见
mindspore.ops.sub()
。说明
当两个输入具有不同的shape时,它们的shape必须要能广播为一个共同的shape。
两个输入不能同时为bool类型。[True, Tensor(True, bool_), Tensor(np.array([True]), bool_)]等都为bool类型。
两个输入遵循隐式类型转换规则,使数据类型保持一致。
- 输入:
- 输出:
Tensor,shape与两个输入广播后的shape相同,数据类型为两个输入中精度较高的类型。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> 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]