为了账号安全,请及时绑定邮箱和手机立即绑定

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

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

弑天下 2022-11-29 17:11:48
抱歉标题措辞不佳。我想做的是这样的:  矩阵 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 列开始。


查看完整回答
反对 回复 2022-11-29
  • 1 回答
  • 0 关注
  • 84 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
微信客服

购课补贴
联系客服咨询优惠详情

帮助反馈 APP下载

慕课网APP
您的移动学习伙伴

公众号

扫描二维码
关注慕课网微信公众号