Fonction torch.aminmax de PyTorch

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

torch.aminmaxC'est une fonction de PyTorch utilisée pour renvoyer les valeurs minimale et maximale d'un tenseur.

Définition de fonction

torch.aminmax(input, dim=None, keepdim=False)

Exemple d'utilisation

Exemple

import torch

x = torch.tensor([[1, 3, 2], [4, 1, 3]])

# Renvoyer les valeurs minimale et maximale de tous les éléments
mm = torch.aminmax(x)
print("Valeur minimale :", mm[0])
print("Valeur maximale :", mm[1])

# Valeurs minimale et maximale selon dim=0
mm0 = torch.aminmax(x, dim=0)
print("minimum selon dim=0 :", mm0[0])
print("maximum selon dim=0 :", mm0[1])

# Valeurs minimale et maximale selon dim=1
mm1 = torch.aminmax(x, dim=1)
print("minimum selon dim=1 :", mm1[0])
print("maximum selon dim=1 :", mm1[1])

La sortie est :

最小值: tensor(1)
最大值: tensor(4)
dim=0 最小值: tensor([1, 1, 2])
dim=0 最大值: tensor([4, 3, 3])
dim=1 最小值: tensor([1, 1])
dim=1 最大值: tensor([3, 4])

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

Autres extensions