Как сопоставить значения из 3D-тензора в 1D-тензор в PyTorch? ⇐ Python

[Anonymous]

Я застрял с проблемой Pytorch и могу использовать некоторую помощь: < /p>
У меня есть два тензора: < /p>

3D тензор (форма: i, j, j) с целочисленными значениями от 0 до n < /li>
1d тензора (форма: n) < /li>
< /ul>
Мне нужно создать новый тензор, который является той же формой, что и первый (i, j, j), но где каждое значение заменяется соответствующим значением со второго тензора. Как использование значений в первом тензоре в качестве индексов для второго. ">

Код: Выделить всё

import torch

# My tensors
big_tensor = torch.randint(0, 256, (10, 25, 25))
small_tensor = torch.rand(256)

# What I'm trying to do
result = magic_function(big_tensor, small_tensor)

# How it should work
print(big_tensor[0, 0, 0])    # Let's say this outputs 42
print(small_tensor[42])       # This might output 0.7853
print(result[0, 0, 0])        # This should also be 0.7853

Я ищу эффективный способ сделать это, желательно без использования циклов, поскольку оба тензора могут быть довольно большими. Есть ли эффективная операция или метод Pytorch, который я могу использовать для этого?
Любая помощь будет замечательной — заранее спасибо!

Подробнее здесь: https://stackoverflow.com/questions/793 ... in-pytorch