Advertisement

tensor.view() 函数的理解

阅读量:

引言:首先简单说明一下,我们会在什么场景中接触到这个函数呢?最常见的情况是在使用pytorch库构建神经网络模型的过程中,这个函数会频繁出现。如果不了解它,你可能会感到困惑,甚至在未来的学习过程中也会持续遇到障碍。如果大家有时间的话,不妨一起来探讨一下这个函数的作用。

1.tensor.view()函数。

(装逼)这个函数本质上也是一种调整尺寸的功能,如果用通俗的语言来理解,它就类似于reshape操作,只不过这里是对张量进行重塑,即将张量的维度(形状大小)重新设置为用户期望的形式。不过我仍然有些模糊(说了这么多还是没能完全理解),为了更清晰地说明问题,下面举一个具体的例子:

代码:

复制代码
>       1. data1=torch.randn((3,2))

>  
>       2. print('data1的数据类型:',type(data1))
>  
>       3. print('data1值为:',data1)
>  
>       4.  
>  
>       5. data2=data1.view(2,3)
>  
>       6. print('data2的数据类型:',type(data2))
>  
>       7. print('data2值为:',da

全部评论 (0)

还没有任何评论哟~