是否有一个 PyTorch 张量函数可以减少规则模式中的维度

Question

抱歉标题措辞不佳。我想做的是这样的：  矩阵 1 是原始矩阵，矩阵 2 是矩阵 1，但每个偶数列和行都被取出。矩阵 3 是矩阵 1，但只有 1 (mod 3) 列和行。矩阵 4 相同，具有 1 (mod 4) 列和行。矩阵 5 有 1 (mod 2) 列和所有行。

是否有以这种方式快速且可以利用 GPU 来操纵张量的 PyTorch 函数？这有点像 MaxPool2d，但我只需要第一个值而不是最大值。如果没有这样的功能，有没有办法手动完成但速度仍然很快？

Answer 1

Matrix 5 是最容易展示的，因为你只需要沿着一个维度切片。但是您可以将两者切分以获得其他结果。matrix5 = matrix1[, ::2]这种表示法每隔一列取一次，从第 0 列开始。