torch.select

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

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

Примечание

Если input является разреженным тензором и вернуть его представление невозможно, возникает исключение RuntimeError. В таком случае рассмотрите возможность использования функции torch.select_copy().

Параметры:

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

Примечание

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