0
点赞
收藏
分享

微信扫一扫

十、pytorch之张量降维(4维降3维)


x1 = [8,32,256,256]
x1 = x1.permute((0, 2, 3, 1)).flatten(1, 2)

x1.permute((0, 2, 3, 1)):[8,32,256,256]—>[8,256,256,32] 维度调转
flatten(1, 2):[8,256,256,32] —>[8,65536,32] 2、3维度平摊为1维,由4维数据降为3维数据


举报

相关推荐

0 条评论