torch.select

torch.select(input, dim, index) → Tensor

Вырезает фрагмент тензора input по выбранному измерению в заданном индексе. Эта функция возвращает представление исходного тензора с удалённым заданным измерением.

Параметры:

• input (Tensor) – входной тензор.
• dim (int) – измерение для среза
• index (int) – индекс для выбора

Примечание

select() эквивалентно срезу. Например, tensor.select(0, index) эквивалентно tensor[index] и tensor.select(2, index) эквивалентно tensor[:,:,index].