Shortcuts

torch.ge

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

Computes inputother\text{input} \geq \text{other} element-wise.

The second argument can be a number or a tensor whose shape is broadcastable with the first argument.

Parameters
  • input (Tensor) – the tensor to compare

  • other (Tensor or float) – the tensor or value to compare

Keyword Arguments

out (Tensor, optional) – the output tensor.

Returns

A boolean tensor that is True where input is greater than or equal to other and False elsewhere

Example:

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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources