PyTorch
torch / torch
torch.isposinf¶
-
torch.
isposinf
(input, *, out=None) → Tensor¶ Tests if each element of
input
is positive infinity or not.- Parameters
input (Tensor) – the input tensor.
- Keyword Arguments
out (Tensor, optional) – the output tensor.
- Example::
>>> a = torch.tensor([-float('inf'), float('inf'), 1.2]) >>> torch.isposinf(a) tensor([False, True, False])
此页内容是否对您有帮助
感谢反馈!