Función torch.unravel_index de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.unravel_indexes una función en PyTorch que se utiliza para expandir índices aplanados a índices multidimensionales. En contrastetorch.ravel_indexcon esto, convierte una posición en un arreglo aplanado de vuelta a coordenadas en un arreglo multidimensional.

Definición de la función

torch.unravel_index(indices, shape)

Descripción de parámetros

  • indices: índice aplanado
  • shape: forma del arreglo multidimensional

Ejemplo de uso

Ejemplo

import torch

# Índice aplanado
indices = torch.tensor([0, 1, 5, 6, 7])

# Forma del arreglo
shape = (2, 4)

# Expandir a índices multidimensionales
result = torch.unravel_index(indices, shape)

print("Índice aplanado:", indices)
print("Forma del arreglo:", shape)
print("Índice multidimensional:")
print(result)

El resultado es:

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

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones