Отбор элементов в PyTorch

Иногда нужно взять не один элемент и не срез, а целый набор ячеек по заранее заданным правилам. Один приём перечисляет номера рядов или столбцов вдоль выбранной оси и собирает из них новую таблицу.

Другой путь опирается на логическую маску той же формы, что и данные: отмеченные ячейки можно вытянуть в один ряд или, наоборот, заменить в исходной таблице одним числом, не трогая остальные значения.

Отдельно встречается сбор, когда для каждой позиции результата задан свой номер внутри исходного тензора - так получают таблицу той же формы, что и набор индексов.

Если нужно только перечислить, где лежат ненулевые значения, удобно получить таблицу координат таких ячеек.

Со всем этим мы и будем разбираться в рамках данного раздела.