mindspore.ops.ge
- mindspore.ops.ge(input, other)[源代码]
输入两个数据,逐元素比较第一个数据是否大于等于第二个数据。
说明
输入 input 和 other 遵循隐式类型转换规则,使数据类型保持一致。
输入必须是两个Tensor,或一个Tensor和一个Scalar。
当输入是两个Tensor时,它们的数据类型不能同时是bool,并保证其shape可以广播。
当输入是一个Tensor和一个Scalar时,Scalar只能是一个常数。
支持广播。
若输入的Tensor可以广播,则会把低维度通过复制该维度的值的方式扩展到另一个输入中对应的高维度。
\[\begin{split}out_{i} =\begin{cases} & \text{True, if } input_{i}>=other_{i} \\ & \text{False, if } input_{i}<other_{i} \end{cases}\end{split}\]- 参数:
input (Union[Tensor, Number, bool]) - 第一个输入可以是Number,也可以是数据类型为Number的Tensor。
other (Union[Tensor, Number, bool]) - 第二个输入是Number,当第一个输入是Tensor时,也可以是bool,或数据类型为Number或bool的Tensor。
- 返回:
Tensor,输出的shape与输入广播后的shape相同,数据类型为bool。
- 异常:
TypeError - input 和 other 都不是Tensor。
- 支持平台:
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([1, 1, 4]), mindspore.int32) >>> output = ops.ge(x, y) >>> print(output) [True True False]