Fonction torch.fmax de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.fmaxIl s'agit d'une fonction dans PyTorch pour le maximum élément par élément (en ignorant NaN). Lorsque l'entrée contient NaN, torch.maximum renvoie NaN, tandis que torch.fmax ignore NaN et renvoie des valeurs non-NaN.

Définition de fonction

torch.fmax(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])

# Maximum élément par élément (en ignorant NaN)
result = torch.fmax(x, y)
print(result)

Le résultat de sortie est:

tensor([3., 5., 4.])

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions