Fonction torch.minimum de PyTorch
Manuel de référence Pytorch torch
torch.minimumC'est une fonction de PyTorch utilisée pour prendre la valeur minimale élément par élément. Elle renvoie un nouveau tenseur composé des valeurs minimales des éléments correspondants de deux tenseurs.
Définition de fonction
torch.minimum(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 minimum élément par élément
result = torch.minimum(x, y)
print(result)
# Créer deux tenseurs
x = torch.tensor([3, 1, 4])
y = torch.tensor([1, 5, 2])
# Prendre le minimum élément par élément
result = torch.minimum(x, y)
print(result)
Le résultat de sortie est:
tensor([1, 1, 2])
Autres extensions