Fonction torch.topk de PyTorch

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

torch.topkC'est une fonction de PyTorch utilisée pour renvoyer les k plus grands éléments et leurs indices.

Définition de fonction

torch.topk(input, k, dim, largest, sorted, out)

Exemple d'utilisation

Exemple

import torch

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

# Renvoyer les 2 plus grands éléments
values, indices = torch.topk(x, k=2)

print("Les 2 plus grandes valeurs :")
print(values)
print("Indices correspondants :")
print(indices)

Le résultat de sortie est:

最大的 2 个值:
tensor([[3, 2],
        [6, 5]])
对应的索引:
tensor([[0, 2],
        [0, 2]])

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

Autres extensions