torch.isposinf — PyTorch 2.7 documentation (original) (raw)

• Docs >
• torch >
• torch.isposinf
• 

Shortcuts

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])

• torch.isposinf
  • isposinf()