PyTorch中permute的基本用法示例
更新时间：2022年04月26日 11:10:10 作者：York1996
pytorch中的permute就像是numpy中的transpose()函数一样,根据指定的维度进行转置,下面这篇文章主要给大家介绍了关于PyTorch中permute的基本用法,需要的朋友可以参考下
permute(dims)
将tensor的维度换位。
参数：参数是一系列的整数，代表原来张量的维度。比如三维就有0，1，2这些dimension。
例：
import torch import numpy as np a=np.array([[[1,2,3],[4,5,6]]]) unpermuted=torch.tensor(a) print(unpermuted.size()) # ——> torch.Size([1, 2, 3]) permuted=unpermuted.permute(2,0,1) print(permuted.size()) # ——> torch.Size([3, 1, 2])
再比如图片img的size比如是（28，28，3）就可以利用img.permute(2,0,1)得到一个size为（3，28，28）的tensor。
利用这个函数permute（0，2，1）可以把Tensor([[[1,2,3],[4,5,6]]]) 转换成
tensor([[[1., 4.],
[2., 5.],
[3., 6.]]])如果使用view，可以得到
tensor([[[1., 2.],
[3., 4.],
[5., 6.]]])关于view的用法：参见PyTorch中view的用法
附：permute(多维数组,[维数的组合])
比如:
a=rand(2,3,4); %这是一个三维数组，各维的长度分别为：2，3，4
%现在交换第一维和第二维：
permute(A,[2,1,3]) %变成3*2*4的矩阵
import torch
import numpy as np
a=np.array([[[1,2,3],[4,5,6]]])
unpermuted=torch.tensor(a)
print(unpermuted.size()) # ——> torch.Size([1, 2, 3])
tensor([[[1., 4.],
[2., 5.],
[3., 6.]]])
permuted=unpermuted.permute(2,0,1)
print(permuted.size()) # ——> torch.Size([3, 1, 2])
tensor([[[1., 2.],
[3., 4.],
[5., 6.]]])
总结
到此这篇关于PyTorch中permute的基本用法的文章就介绍到这了,更多相关PyTorch permute的用法内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了关于python的第三方库下载与更改方式,使用python的朋友都知道python有很多非常方便的第三方库可以使用,那么如果下载这些第三方库呢,今天小编就带你们来看看2023-04-04
因为买了阿里/腾讯的云服务器,但是使用云存储还需要收费,又加上家里正好有一台nas,又加上闲的没事,所以搞了一个小脚本,这个项目主要功能是为typora增加一个自定义图床,本文给大家介绍基于python tornado实现图床功能,感兴趣的朋友一起看看吧2023-08-08
这篇文章主要介绍了解决pycharm修改代码后第一次运行不生效的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-02-02
这篇文章主要介绍了由Python运算π的值深入Python中科学计算的实现,由简单的计算发散出各种算法的讲解,需要的朋友可以参考下2015-04-04
在numpy的函数库中,np.append()函数是一个常用的数组操作函数,它在进行数组操作时能够将两个数组进行拼接,并返回一个拼接后的新数组,下面就来介绍一下具体用法,感兴趣的可以了解一下2023-11-11
这篇文章主要为大家介绍了Python sqlparse解析SQL表血缘追踪实现，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-10-10
Python中的asyncio模块提供了异步IO支持,通过协程和事件循环实现异步编程,使用装饰器@asyncio.coroutine可以定义协程,yield from语法用于调用其他协程并实现非阻塞等待,asyncio.sleep()模拟IO操作,通过并发执行多个协程提高程序性能2024-10-10
这篇文章主要介绍了Python实现byte转integer操作，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-06-06
Microsoft Word是广泛用于文档编辑和处理的工具,有时候,可能需要从Word文档中提取所有的图片,以进行后续的处理或分析,本文主要介绍了如何使用Python实现这一操作,需要的可以参考下2024-01-01
Python如何将两个三维模型(obj)合成一个三维模型(obj)
这篇文章主要介绍了Python如何将两个三维模型(obj)合成一个三维模型(obj)问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-06-06
最新评论