Fonction torch.fmin de PyTorch
Manuel de référence Pytorch torch
torch.fminIl s'agit d'une fonction dans PyTorch pour le minimum élément par élément (en ignorant NaN). Lorsque l'entrée contient NaN, torch.minimum renvoie NaN, tandis que torch.fmin ignore NaN et renvoie la valeur non-NaN.
Définition de fonction
torch.fmin(input, other)
Exemple d'utilisation
Exemple
import torch
# Créer un tenseur contenant NaN
x = torch.tensor([3.0, float('nan'), 4.0])
y = torch.tensor([1.0, 5.0, 2.0])
# Minimum élément par élément (en ignorant NaN)
result = torch.fmin(x, y)
print(result)
# Créer un tenseur contenant NaN
x = torch.tensor([3.0, float('nan'), 4.0])
y = torch.tensor([1.0, 5.0, 2.0])
# Minimum élément par élément (en ignorant NaN)
result = torch.fmin(x, y)
print(result)
Le résultat de sortie est:
tensor([1., 5., 2.])
Autres extensions