Предположим, у меня есть тензор sequences формы [8, 12, 2].Теперь я хотел бы сделать выбор этого тензора для каждого первого измерения, что приводит к тензору формы [8, 2].Выбор по измерению 1 определяется индексами, хранящимися в длинном тензоре indices формы [8].
Я пробовал это сделать, однако он выбирает каждый индекс в indices для каждого первого измерения в sequencesвместо одного.
sequences[:, indices]
Как сделать этот запрос без медленного и уродливого цикла for?