Fonction torch.max de PyTorch

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

torch.maxIl s'agit d'une fonction de PyTorch utilisée pour calculer la valeur maximale des tenseurs.

Définition de fonction

torch.max(input, dim, keepdim, out)

Exemples d'utilisation

Exemple

import torch

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

# Maximum global
print("Maximum global :", torch.max(x))

# Maximum selon la dimension
print("Maximum pour dim=0 :", torch.max(x, dim=0))

Le résultat de sortie est:

全局最大: tensor(6)
dim=0 最大: torch.return_types.max(values=tensor([4, 5, 6]), indices=tensor([1, 1, 1]))

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

Autres extensions