PyTorch torch.fmax Funktion
PyTorch torch Referenzhandbuch
torch.fmaxEs ist eine Funktion in PyTorch für elementweises Maximum (unter Ignorierung von NaN). Wenn die Eingabe NaN enthält, gibt torch.maximum NaN zurück, während torch.fmax NaN ignoriert und Nicht-NaN-Werte zurückgibt.
Funktionsdefinition
torch.fmax(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 Maximum (unter Ignorierung von NaN)
result = torch.fmax(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 Maximum (unter Ignorierung von NaN)
result = torch.fmax(x, y)
print(result)
Das Ausgabeergebnis ist:
tensor([3., 5., 4.])
Andere Erweiterungen