У меня есть входной тензор для lstm, и его форма является пакетной seq_len emb, и я хочу отсортировать этот тенор по его seq_len dim по другому тензору, форма которого является пакетной * seq_len
например:
есть
[ [ [0,0]
[1,1]
[2,2] ]
[ [0,0]
[1,1]
[2,2] ]
]
и индекс
[[0,1,2]
[2,1,0] ]
желаю, чтобы результат был
[ [ [0,0]
[1,1]
[2,2] ]
[ [2,2]
[1,1]
[0,0] ]
]
Есть ли какая-нибудь необычная операция в pytorch, чтобы сделать это быстро?
спасибо