PyTorch torch.fmin Funktion
PyTorch torch Referenzhandbuch
torch.fminEs ist eine Funktion in PyTorch für elementweises Minimum (unter Ignorierung von NaN). Wenn die Eingabe NaN enthält, gibt torch.minimum NaN zurück, während torch.fmin NaN ignoriert und den Nicht-NaN-Wert zurückgibt.
Funktionsdefinition
torch.fmin(input, other)
Anwendungsbeispiel
Beispiel
import torch
# Einen Tensor erstellen, der NaN enthält
x = torch.tensor([3.0, float('nan'), 4.0])
y = torch.tensor([1.0, 5.0, 2.0])
# Elementweises Minimum (unter Ignorierung von NaN)
result = torch.fmin(x, y)
print(result)
# Einen Tensor erstellen, der NaN enthält
x = torch.tensor([3.0, float('nan'), 4.0])
y = torch.tensor([1.0, 5.0, 2.0])
# Elementweises Minimum (unter Ignorierung von NaN)
result = torch.fmin(x, y)
print(result)
Das Ausgabeergebnis ist:
tensor([1., 5., 2.])
Andere Erweiterungen