当前位置 : 主页 > 编程语言 > python >

Pytorch函数view、permute、squeeze、usqueeze

来源:互联网 收集:自由互联 发布时间:2022-06-24
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)

如下例所示
Pytorch函数view、permute、squeeze、usqueeze_pytorch
Pytorch函数view、permute、squeeze、usqueeze_python_02
Pytorch函数view、permute、squeeze、usqueeze_数据_03
Pytorch函数view、permute、squeeze、usqueeze_深度学习_04

总结一下:view的二种方法,一种就是当成resize一样转换数据shape,一种是如果是torch.view(参数a,-1),则表示在参数b未知,参数a已知的情况下自动补齐列向量长度,在这个例子中a=3,x总共由9个元素,则b=9/3=3。以此类推更多维度的比如说

上一篇:最全的数据科学常用pandas教程
下一篇:没有了
网友评论