slice

Вопрос или проблема Я хочу сделать срез 3D тензора в PyTorch. Размер 3D тензора src_tensor равен (batch, max_len, hidden_dim), и у меня есть 1D вектор индексов indices с размером (batch,). Я хочу сделать срез вдоль второго измерения src_tensor.