Fonction torch.maximum de PyTorch
Manuel de référence Pytorch torch
torch.maximumC'est une fonction dans PyTorch pour le maximum élément par élément. Renvoie un nouveau tenseur composé du maximum des éléments correspondants de deux tenseurs.
Définition de fonction
torch.maximum(input, other)
Exemple d'utilisation
Exemple
import torch
# Créer deux tenseurs
x = torch.tensor([3, 1, 4])
y = torch.tensor([1, 5, 2])
# Prendre le maximum élément par élément
result = torch.maximum(x, y)
print(result)
# Créer deux tenseurs
x = torch.tensor([3, 1, 4])
y = torch.tensor([1, 5, 2])
# Prendre le maximum élément par élément
result = torch.maximum(x, y)
print(result)
Le résultat de sortie est:
tensor([3, 5, 4])
Autres extensions