Функция PyTorch torch.unravel_index
Справочное руководство PyTorch torch
torch.unravel_index— это функция в PyTorch, используемая для разворачивания плоских индексов в многомерные индексы. В отличие отtorch.ravel_index, наоборот, преобразует позицию в плоском массиве обратно в координаты в многомерном массиве.
Определение функции
torch.unravel_index(indices, shape)
Описание параметра
indices: Плоский индексshape: Форма многомерного массива
Пример использования
Пример
import torch
# Плоский индекс
indices = torch.tensor([0, 1, 5, 6, 7])
# Форма массива
shape = (2, 4)
# Развернуть в многомерный индекс
result = torch.unravel_index(indices, shape)
print("Плоский индекс:", indices)
print("Форма массива:", shape)
print("Многомерный индекс:")
print(result)
# Плоский индекс
indices = torch.tensor([0, 1, 5, 6, 7])
# Форма массива
shape = (2, 4)
# Развернуть в многомерный индекс
result = torch.unravel_index(indices, shape)
print("Плоский индекс:", indices)
print("Форма массива:", shape)
print("Многомерный индекс:")
print(result)
Вывод:
展平索引: tensor([0, 1, 5, 6, 7])
数组形状: (2, 4)
多维索引:
tensor([[0, 0, 1, 1, 1],
[0, 1, 1, 0, 1]])
Другие расширения