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)
还没有任何评论哟~
