torch.ge

torch.ge(input, other, *, out=None) → Tensor

计算元素级 $\text{input} \geq \text{other}$

第二个参数可以是与第一个参数可广播的数字或张量。

参数:

• 输入（张量）- 要比较的张量

• 其他（张量或浮点数）- 要比较的张量或值

关键字参数:

输出（张量，可选）- 输出张量。

返回值:

一个布尔张量，当 input 大于或等于 other 时为 True，其他地方为 False

示例：

>>> torch.ge(torch.tensor([[1, 2], [3, 4]]), torch.tensor([[1, 1], [4, 4]]))
tensor([[True, True], [False, True]])