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

• Docs >
• torch >
• torch.isneginf
• 

Shortcuts

torch.isneginf(input, *, out=None) → Tensor¶

Tests if each element of input is negative 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.isneginf(a) tensor([ True, False, False])

• torch.isneginf
  • isneginf()