Fonction torch.amax de PyTorch

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

torch.amaxIl s'agit d'une fonction de PyTorch utilisée pour renvoyer la valeur maximale d'un tenseur le long de la dimension spécifiée.

Définition de fonction

torch.amax(input, dim, keepdim=False)

Exemples d'utilisation

Exemple

import torch

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

# Renvoyer la valeur maximale de tous les éléments
print("Maximum global :", torch.amax(x))

# Maximum selon dim=0
print("maximum dim=0 :", torch.amax(x, dim=0))

# Maximum selon dim=1
print("maximum dim=1 :", torch.amax(x, dim=1))

Le résultat de sortie est:

全局最大: tensor(4)
dim=0 最大: tensor([4, 3, 3])
dim=1 最大: tensor([3, 4])

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

Autres extensions