分类：压缩Tensor维度

2020
10-10

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：要对哪个维度扩展就输入那个维度的... 继续阅读 >