【发布时间】：2020-05-21 03:02:14
【问题描述】：
很抱歉标题的措辞不佳。我想做的是这样的： 矩阵 1 是原始矩阵，矩阵 2 是矩阵 1，但每个偶数列和行都被取出。矩阵 3 是矩阵 1，但只有 1 (mod 3) 列和行。矩阵 4 是相同的，具有 1（模 4）列和行。矩阵 5 有 1 (mod 2) 列和所有行。
是否有一个 PyTorch 函数可以以这种快速且可以利用 GPU 的方式操纵张量？这有点像 MaxPool2d，但是我只需要第一个值而不是最大值。如果没有类似的功能，有没有办法手动完成但仍然很快？
【问题讨论】：
-
这不就是切片吗？
-
是的，但我不想有 for 循环并选择单个列和行，因为这会很慢。我正在寻找可以快速执行此操作的内部函数
-
我知道。使用切片。这不会涉及 for 循环。