pytorch中的view和view逆操作

将x4平铺成一维向量，得到x5；

将x5压缩成和x4shape一样，得到x6

x5 = x4.view(x4.size(0), -1)
x6 = x5.detach().view_as(x4)

 

posted @ 2021-03-29 11:18  皮卡皮卡妞  阅读(184)  评论(0)  编辑  收藏  举报