torch.select

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

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

Примечание

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

Параметры

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

Примечание

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