PyTorch
torch / torch
torch.gt¶
-
torch.
gt
(input, other, *, out=None) → Tensor¶ Computes input>other element-wise.
The second argument can be a number or a tensor whose shape is broadcastable with the first argument.
- Parameters
- Keyword Arguments
out (Tensor, optional) – the output tensor.
- Returns
A boolean tensor that is True where
input
is greater thanother
and False elsewhere
Example:
>>> torch.gt(torch.tensor([[1, 2], [3, 4]]), torch.tensor([[1, 1], [4, 4]])) tensor([[False, True], [False, False]])
此页内容是否对您有帮助
感谢反馈!