Извините за плохую формулировку названия. Я хотел сделать вот что: 
Матрица 1 - это исходная матрица, а матрица 2 - это матрица 1, но все четные столбцы и строки удалены. Матрица 3 является матрицей 1, но имеет только 1 (mod 3) столбец и строку. Матрица 4 такая же, с 1 столбцом и строкой (по модулю 4). Матрица 5 имеет 1 столбец (по модулю 2) и все строки.
Есть ли функция PyTorch, которая управляет тензорами таким образом, быстро и может использовать графический процессор? Это вроде как MaxPool2d, но мне нужно только первое значение, а не максимальное. Если таких функций нет, есть ли способ сделать это вручную, но при этом быстро?
