1.view 在pytorch中view函数的作用为重构张量的维度,相当于numpy中resize()的功能,但是用法可能不太一样。 view有二种用法: torch.view(参数a,参数b,。。。) torch.view(-1)或torch.view(参数a,
1.view
在pytorch中view函数的作用为重构张量的维度,相当于numpy中resize()的功能,但是用法可能不太一样。
view有二种用法:
- torch.view(参数a,参数b,。。。)
- torch.view(-1)或torch.view(参数a,-1)
如下例所示
总结一下:view的二种方法,一种就是当成resize一样转换数据shape,一种是如果是torch.view(参数a,-1),则表示在参数b未知,参数a已知的情况下自动补齐列向量长度,在这个例子中a=3,x总共由9个元素,则b=9/3=3。以此类推更多维度的比如说