Fonction torch.unravel_index de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.unravel_indexest une fonction de PyTorch utilisée pour dérouler des indices aplatis en indices multidimensionnels. Contrairement àtorch.ravel_indexinversement, elle convertit une position dans un tableau aplati en coordonnées dans un tableau multidimensionnel.

Définition de fonction

torch.unravel_index(indices, shape)

Description du paramètre

  • indices: Indice aplati
  • shape: Forme du tableau multidimensionnel

Exemple d'utilisation

Exemple

import torch

# Indice aplati
indices = torch.tensor([0, 1, 5, 6, 7])

# Forme du tableau
shape = (2, 4)

# Dérouler en indice multidimensionnel
result = torch.unravel_index(indices, shape)

print("Indice aplati:", indices)
print("Forme du tableau:", shape)
print("Indice multidimensionnel:")
print(result)

La sortie est :

展平索引: tensor([0, 1, 5, 6, 7])
数组形状: (2, 4)
多维索引:
tensor([[0, 0, 1, 1, 1],
        [0, 1, 1, 0, 1]])

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions