Fonction torch.min de PyTorch
Manuel de référence Pytorch torch
torch.minIl s'agit d'une fonction de PyTorch utilisée pour calculer la valeur minimale des tenseurs.
Définition de fonction
torch.min(input, dim, keepdim, out)
Exemple d'utilisation
Exemple
import torch
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
print(Minimum global :, torch.min(x))
print(minimum pour dim=0 :, torch.min(x, dim=0))
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
print(Minimum global :, torch.min(x))
print(minimum pour dim=0 :, torch.min(x, dim=0))
Autres extensions