原文地址:http://www.mamicode.com/info-detail-2561432.html
unsqueeze(arg)是增添第arg个维度为1,以插入的形式填充
squeeze(arg)是删除第arg个维度(如果当前维度不为1,则不会进行删除)
torch版本的降维和升维
最新推荐文章于 2022-11-16 19:47:08 发布
原文地址:http://www.mamicode.com/info-detail-2561432.html
unsqueeze(arg)是增添第arg个维度为1,以插入的形式填充
squeeze(arg)是删除第arg个维度(如果当前维度不为1,则不会进行删除)