pytorch中tensor的squeeze()和unsqueeze()

squeeze()：

squeeze(arg)表示第arg维的维度值为1，则去掉该维度。否则tensor不变。（即若tensor.shape()[arg] = 1，则去掉该维度）

unsqueeze()：

unsqueeze(arg)与squeeze(arg)作用相反，表示在第arg维增加一个维度值为1的维度。

常用操作：

参考：https://blog.csdn.net/lanse_zhicheng/article/details/79148678

转载于:https://www.cnblogs.com/sbj123456789/p/9231571.html