Fonction torch.kthvalue de PyTorch
Manuel de référence PyTorch torch
torch.kthvalueC'est une fonction dans PyTorch utilisée pour renvoyer la k-ième plus petite valeur. Elle renvoie la k-ième plus petite valeur et son indice le long de la dimension spécifiée du tenseur d'entrée.
Définition de fonction
torch.kthvalue(input, k, dim=None, keepdim=False)
Exemple d'utilisation
Exemple
import torch
# Créer un tenseur
x = torch.tensor([3, 1, 2, 4])
# Renvoyer la 2e plus petite valeur et l'indice
values, indices = torch.kthvalue(x, 2)
print(values)
print(indices)
# Créer un tenseur
x = torch.tensor([3, 1, 2, 4])
# Renvoyer la 2e plus petite valeur et l'indice
values, indices = torch.kthvalue(x, 2)
print(values)
print(indices)
Le résultat de sortie est:
tensor(2) tensor(2)
Autres extensions