157239n
抱歉,标题措辞不佳。我想做的是这样的:矩阵1是原始矩阵,矩阵2是矩阵1,但取出的列和行均为偶数。矩阵3是矩阵1,但只有1(模3)列和行。矩阵4相同,具有1(mod 4)列和行。矩阵5具有1(模2)列和所有行。
是否有PyTorch函数以这种方式操纵张量,速度很快并且可以利用GPU?这有点像MaxPool2d,但是我只需要第一个值而不是最大值。如果没有类似的功能,有没有办法手动但仍然快速?
艾莉亚·麦卡锡(Arya McCarthy)
矩阵5最容易显示,因为您只需要沿一个维度进行切片。但是您可以同时切片以获取其他结果。
matrix5 = matrix1[, ::2]
此表示法从第二个列开始,每隔一列。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
编辑于
我来说两句