Advertisement

torch.tensor.permute() (《pytorch学习》)

阅读量:

引言:从个人理解来看,该函数的功能主要涉及张量维度的转换以及张量矩阵的转置操作。

1.张量维度变换。

根据官方文档的说明,其形式为:Tensor.``permute(*dims) → Tensor(然而对于这一概念的理解仍存在一定的困惑)

随后查阅了这里的相关内容,并结合自身思考进行了一番探索!以下将通过具体示例进行说明!

代码:

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

>  
>       2. print('data1的数据类型:',type(data1))
>  
>       3. print('data1的数据维度:',data1.shape)
>  
>       4.  
>  
>       5. data2=data1.permute(2,1,0)
>  
>       6. print('data2的数据类型:',type(data2))
>  
>       7. print('data2的数据维度:',data2.shape)
>  
>  
>  
>  
> ```

>
>
>
> output:
>
>
>
> data1的数据类型: <class 'torch.Tensor

全部评论 (0)

还没有任何评论哟~