抱歉标题措辞不佳。我想做的是这样的: 矩阵 1 是原始矩阵,矩阵 2 是矩阵 1,但每个偶数列和行都被取出。矩阵 3 是矩阵 1,但只有 1 (mod 3) 列和行。矩阵 4 相同,具有 1 (mod 4) 列和行。矩阵 5 有 1 (mod 2) 列和所有行。是否有以这种方式快速且可以利用 GPU 来操纵张量的 PyTorch 函数?这有点像 MaxPool2d,但我只需要第一个值而不是最大值。如果没有这样的功能,有没有办法手动完成但速度仍然很快?
1 回答

繁星点点滴滴
TA贡献1803条经验 获得超3个赞
Matrix 5 是最容易展示的,因为你只需要沿着一个维度切片。但是您可以将两者切分以获得其他结果。
matrix5 = matrix1[, ::2]
这种表示法每隔一列取一次,从第 0 列开始。
添加回答
举报
0/150
提交
取消