PyTorch torch.topk-Funktion
PyTorch torch-Referenzhandbuch
torch.topkEs ist eine Funktion in PyTorch, die verwendet wird, um die größten k Elemente und ihre Indizes zurückzugeben.
Funktionsdefinition
torch.topk(input, k, dim, largest, sorted, out)
Anwendungsbeispiel
Beispiel
import torch
x = torch.tensor([[3, 1, 2], [6, 4, 5]])
# Die größten 2 Elemente zurückgeben
values, indices = torch.topk(x, k=2)
print("Die größten 2 Werte:")
print(values)
print("Entsprechende Indizes:")
print(indices)
x = torch.tensor([[3, 1, 2], [6, 4, 5]])
# Die größten 2 Elemente zurückgeben
values, indices = torch.topk(x, k=2)
print("Die größten 2 Werte:")
print(values)
print("Entsprechende Indizes:")
print(indices)
Das Ausgabeergebnis ist:
最大的 2 个值:
tensor([[3, 2],
[6, 5]])
对应的索引:
tensor([[0, 2],
[0, 2]])
Andere Erweiterungen