PyTorch torch.unravel_index-Funktion
PyTorch torch-Referenzhandbuch
torch.unravel_indexist eine Funktion in PyTorch, die verwendet wird, um abgeflachte Indizes in mehrdimensionale Indizes zu entwirren. Anders alstorch.ravel_indexumgekehrt wandelt sie eine Position in einem abgeflachten Array zurück in Koordinaten in einem mehrdimensionalen Array um.
Funktionsdefinition
torch.unravel_index(indices, shape)
Parameterbeschreibung
indices: Abgeflachter Indexshape: Form des mehrdimensionalen Arrays
Anwendungsbeispiel
Beispiel
import torch
# Abgeflachter Index
indices = torch.tensor([0, 1, 5, 6, 7])
# Form des Arrays
shape = (2, 4)
# In einen mehrdimensionalen Index entwirren
result = torch.unravel_index(indices, shape)
print("Abgeflachter Index:", indices)
print("Form des Arrays:", shape)
print("Mehrdimensionaler Index:")
print(result)
# Abgeflachter Index
indices = torch.tensor([0, 1, 5, 6, 7])
# Form des Arrays
shape = (2, 4)
# In einen mehrdimensionalen Index entwirren
result = torch.unravel_index(indices, shape)
print("Abgeflachter Index:", indices)
print("Form des Arrays:", shape)
print("Mehrdimensionaler Index:")
print(result)
Die Ausgabe ist:
展平索引: tensor([0, 1, 5, 6, 7])
数组形状: (2, 4)
多维索引:
tensor([[0, 0, 1, 1, 1],
[0, 1, 1, 0, 1]])
Andere Erweiterungen