pytorch squeeze()和unsqueeze()
squeeze():
squeeze(arg)表示第arg维的维度值为1,则去掉该维度。否则tensor不变。(即若tensor.shape()[arg] = 1,则去掉该维度)
unsqueeze():
unsqueeze(arg)与squeeze(arg)作用相反,表示在第arg维增加一个维度值为1的维度。
常用操作:
总结:
squeeze和unsqueeze都不会增删数据本身,只是改变矩阵的维度,然后把被删维度的数据放到未删的维度上
参考:https://blog.csdn.net/lanse_zhicheng/article/details/79148678