Как сопоставить значения из 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
< /code>
Я ищу исполнительный способ сделать это, предпочтительно не использовать петли, так как оба тензора могут быть довольно большими. Есть ли эффективная операция или метод Pytorch, который я могу использовать для этого?
Любая помощь была бы потрясающей - заранее спасибо!

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