Я хочу проиндексировать тензор следующим образом:
A = [[4,5], [6,7]], index = [[0,2], [2,1]] - (?) -> [[nan, nan, 6], [4, 7, 5]].
По сути, элемент индекса - это столбец в выходном массиве, соответствующийэлемент в A принадлежит, в то время как индекс строки просто складывается на основе числа вхождений номера столбца в индексе.
Можно ли преобразовать это значение в numpy, выполнить преобразование и преобразовать обратно в тензор всередина моего конвейера ML, как NumPy имеет более сложную индексацию массива?Могу ли я просто сделать это с помощью pytorch?