脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → PyTorch permute的用法

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提取Linux内核源代码的目录结构实现方法

    Python提取Linux内核源代码的目录结构实现方法

    下面小编就为大家带来一篇Python提取Linux内核源代码的目录结构实现方法。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2016-06-06
  • Pytest测试报告工具Allure用法介绍

    Pytest测试报告工具Allure用法介绍

    这篇文章介绍了Pytest测试报告工具Allure的用法，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2022-07-07
  • Python基于TensorFlow接口实现深度学习神经网络回归

    Python基于TensorFlow接口实现深度学习神经网络回归

    这篇文章主要为大家详细介绍了如何基于Python语言中TensorFlow的tf.estimator接口，实现深度学习神经网络回归的具体方法，感兴趣的可以了解一下
    2023-02-02
  • Python2和Python3中print的用法示例总结

    Python2和Python3中print的用法示例总结

    在Python 3中接触的第一个很大的差异就是缩进是作为语法的一部分，这和C++等其他语言确实很不一样，所以要小心，其中python3和python2中print的用法有很多不同，这篇文章主要给大家介绍了关于Python2和Python3中print用法的相关资料，需要的朋友可以参考下。
    2017-10-10
  • Python 动态变量名定义与调用方法

    Python 动态变量名定义与调用方法

    这篇文章主要介绍了Python 动态变量名定义与调用方法,需要的朋友可以参考下
    2020-02-02
  • Python实现服务端渲染SSR的示例代码

    Python实现服务端渲染SSR的示例代码

    服务端渲染是一种常见的技术策略,特别是在需要改善网站的搜索引擎优化（SEO）和首屏加载时间的场景下,本文将介绍如何利用 Python 实现 SSR,感兴趣的可以了解下
    2024-02-02
  • Python中读取文件名中的数字的实例详解

    Python中读取文件名中的数字的实例详解

    在本篇文章里小编给大家整理了一篇关于Python中读取文件名中的数字的实例详解内容，有兴趣的朋友们可以参考下。
    2020-12-12
  • Python实现敲击木鱼积累功德小项目

    Python实现敲击木鱼积累功德小项目

    最近大家都很流行用手机敲击电子木鱼积累功德，这在很多短视频中也常常见到。本文将用Python实现这一效果，感兴趣的小伙伴可以了解一下
    2022-11-11
  • OpenCV4.1.0+VS2017环境配置的方法步骤

    OpenCV4.1.0+VS2017环境配置的方法步骤

    这篇文章主要介绍了OpenCV4.1.0+VS2017环境配置的方法步骤，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-07-07
  • 在python中写个自定义数据包协议的打包和解包测试

    在python中写个自定义数据包协议的打包和解包测试

    这篇文章主要介绍了在python中写个自定义数据包协议的打包和解包测试,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2023-09-09

最新评论