PyTorch torch.isinf Funktion
PyTorch torch Referenzhandbuch
torch.isinfist eine Funktion in PyTorch, mit der geprüft wird, ob ein Wert unendlich ist. Gibt einen booleschen Tensor zurück, wobei für jedes Element True gilt, wenn der Wert unendlich ist (positive Unendlichkeit oder negative Unendlichkeit).
Funktionsdefinition
torch.isinf(input)
Anwendungsbeispiel
Beispiel
import torch
# Tensor mit verschiedenen Werten erstellen
x = torch.tensor([1.0, float('inf'), float('-inf'), 2.0])
# Prüfen, ob die Werte unendlich sind
result = torch.isinf(x)
print(result)
# Tensor mit verschiedenen Werten erstellen
x = torch.tensor([1.0, float('inf'), float('-inf'), 2.0])
# Prüfen, ob die Werte unendlich sind
result = torch.isinf(x)
print(result)
Das Ausgabeergebnis ist:
tensor([False, True, True, False])
Andere Erweiterungen