PyTorch torch.isfinite-Funktion
PyTorch torch-Referenzhandbuch
torch.isfiniteDies ist eine Funktion in PyTorch, mit der überprüft wird, ob Tensorelemente endliche Werte sind. Sie gibt einen booleschen Tensor zurück, wobei für jedes Element True gilt, wenn der Wert endlich ist (nicht unendlich oder NaN).
Funktionsdefinition
torch.isfinite(input)
Anwendungsbeispiel
Beispiel
import torch
# Tensor mit verschiedenen Werten erstellen
x = torch.tensor([1.0, float('inf'), float('nan'), 2.0])
# Auf endliche Werte prüfen
result = torch.isfinite(x)
print(result)
# Tensor mit verschiedenen Werten erstellen
x = torch.tensor([1.0, float('inf'), float('nan'), 2.0])
# Auf endliche Werte prüfen
result = torch.isfinite(x)
print(result)
Die Ausgabe ist:
tensor([True, False, False, True])
Andere Erweiterungen