一文掌握torch.squeeze() 和torch.unsqueeze()的用法

torch.squeeze() 这个函数主要对数据的维度进行压缩，去掉维数为1的的维度，默认是将a中全部为1的维度删掉。也能够经过dim指定位置，删掉指定位置的维数为1的维度。python torch.unsqueeze()这个函数主要是对数据维度进行扩充。须要经过dim指定位置，给指定位置加上维数为1的维度。函数 我本身test的代码：code import torch x = torch.z

>>阅读原文<<