PyTorch torch.index_select Funktion
PyTorch torch Referenzhandbuch
torch.index_selectEs ist eine Funktion in PyTorch, die verwendet wird, um Elemente auszuwählen, die den Indizes entlang einer angegebenen Dimension entsprechen.
Funktionsdefinition
torch.index_select(input, dim, index)
Anwendungsbeispiel
Beispiel
import torch
x = torch.randn(4, 5)
# Zeilen 0 und 2 auswählen
indices = torch.tensor([0, 2])
result = torch.index_select(x, dim=0, index=indices)
print("Form des Ergebnisses:", result.shape)
x = torch.randn(4, 5)
# Zeilen 0 und 2 auswählen
indices = torch.tensor([0, 2])
result = torch.index_select(x, dim=0, index=indices)
print("Form des Ergebnisses:", result.shape)
Andere Erweiterungen