PyTorch torch.kthvalue-Funktion
PyTorch torch-Referenzhandbuch
torch.kthvalueEs ist eine Funktion in PyTorch, die verwendet wird, um den k-kleinsten Wert zurückzugeben. Sie gibt den k-kleinsten Wert und seinen Index entlang der angegebenen Dimension des Eingabetensors zurück.
Funktionsdefinition
torch.kthvalue(input, k, dim=None, keepdim=False)
Anwendungsbeispiel
Beispiel
import torch
# Tensor erstellen
x = torch.tensor([3, 1, 2, 4])
# Den 2. kleinsten Wert und Index zurückgeben
values, indices = torch.kthvalue(x, 2)
print(values)
print(indices)
# Tensor erstellen
x = torch.tensor([3, 1, 2, 4])
# Den 2. kleinsten Wert und Index zurückgeben
values, indices = torch.kthvalue(x, 2)
print(values)
print(indices)
Das Ausgabeergebnis ist:
tensor(2) tensor(2)
Andere Erweiterungen