降维和升维

unsqueeze（arg）是增添第arg个维度为1，以插入的形式填充
squeeze（arg）是删除第arg个维度(如果当前维度不为1，则不会进行删除)

posted @ 2021-08-12 20:27  吴莫愁258  阅读(93)  评论(0)  编辑  收藏  举报