主要介绍了Pytorch 扩展Tensor维度、压缩Tensor维度的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

1. 扩展Tensor维度 　　相信刚接触Pytorch的宝宝们，会遇到这样一个问题，输入的数据维度和实验需要维度不一致，输入的可能是2维数据或3维数据，实验需要用到3维或4维数据，那么我们需要扩展这个维度。其实特别简单，只要对数据加一个扩展维度方法就可以了。 1.1torch.unsqueeze(self: Tensor, dim: _int) 　　torch.unsqueeze(self: Tensor, dim: _int) 　　参数说明：self:输入的tensor数据，dim：要对哪个