tensor 和 numpy 的互相转换

tensor 和 numpy 的互相转换为什么要相互转换:1.要对tensor进行操作,需要先启动一个Session,否则,我们无法对一个tensor比如一个tensor常量重新赋值或是做一些判断操作,所以如果将它转化为numpy数组就好处理了。下面一个小程序讲述了将tensor转化为numpy数组,以及又重新还原为tensor:2.Torch的Tensor和numpy的array会共享他们的存储空间,修改一个会导致另外的一…

大家好,又见面了,我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全系列IDE稳定放心使用

为什么要相互转换:

简单一句话, numpy操作多样, 简单. 但网络前向只能是tensor类型, 各有优势, 所以需要相互转换补充.

# convert Tensor x of torch to array y of  numpy: 
y = x.numpy();

# convert array x of  numpy to Tensor y of torch: 
y = torch.from_numpy(x)

# 先将数据转换成Tensor, 再使用CUDA函数来将Tensor移动到GPU上加速

如果想把CUDA tensor格式的数据改成numpy时,需要先将其转换成cpu float-tensor随后再转到numpy格式。 

x_np = x.data.numpy()

# 改为:

x_np = x.data.cpu().numpy()

# 或者兼容上面两者的方式
x_np = x.detach().cpu().numpy() if x.requires_grad else x.cpu().numpy()

参考1:https://blog.csdn.net/ljs_a/article/details/78758116 

参考2:https://blog.csdn.net/u012436149/article/details/54627597

参考3:https://blog.csdn.net/weixin_39449466/article/details/80410314

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请联系我们举报,一经查实,本站将立刻删除。

发布者:全栈程序员-站长,转载请注明出处:https://javaforall.net/180419.html原文链接:https://javaforall.net

(0)
上一篇 2022年10月19日 上午6:36
下一篇 2022年10月19日 上午6:46


相关推荐

  • 六十四卦详细解释_六十四卦断事

    六十四卦详细解释_六十四卦断事文章目录第1卦 乾为天(乾卦) 刚健中正 上上卦第2卦 坤为地(坤卦) 柔顺伸展 上上卦第3卦 水雷屯(屯卦) 起始维艰 下下卦第4卦 山水蒙(蒙卦) 启蒙奋发 中下卦第5卦 水天需(需卦) 守正待机 中上卦第6卦 天水讼(讼卦) 慎争戒讼 中下卦第7卦 地水师(师卦) 行险而顺 中上卦第8卦 水地比(比卦) 诚信团结 上上卦第9卦 风天小畜(小畜卦) 蓄养待进 下下卦第10卦 天泽履(履卦) 脚…

    2022年8月18日
    14
  • MATLAB–相机标定教程

    MATLAB–相机标定教程对相机标定原理不熟悉的同学可以参考张正友标记法的译文MATLAB:2014a待标记图像:http://download.csdn.net/detail/heroacool/9477530应用程序下找到CameraCalibration工具箱加载待标定的图像填写棋盘格每个格子边长的真实值可以预览成功检测出棋盘格的图像,然后开始标定,点击Calibrate平均误差小于0.5即可导出相机

    2022年5月8日
    32
  • EditPlus注册码5.1注册码

    EditPlus注册码5.1注册码Vovan3AG46 JJ48E CEACC 8E6EW ECUAW

    2026年3月18日
    2
  • 注册邮箱发送短信验证_网易邮箱注册系统未收到短信

    注册邮箱发送短信验证_网易邮箱注册系统未收到短信分享概要:讲述yii框架,用户注册然后以邮箱通知和短信通知用户注册状态。短信使用阿里云,邮箱使用swiftmailer插件。支持php~~~功能点:用户注册通知用户注册类:publicfunctionactionCreateUsers(){//数据过滤数据判断这个省略了。。。。直接看重点if($model->save()){//对用户数据的保存…

    2022年10月13日
    5
  • Claude Code中文教程:100%免费,从入门到精通

    Claude Code中文教程:100%免费,从入门到精通

    2026年3月15日
    3
  • pip国内镜像(清华大学镜像)

    网上搜到的pip国内镜像大部分是豆瓣的http://pypi.douban.com/simple/但是根本不全,很多包没有所以推荐清华大学的https://pypi.tuna.tsinghua.edu.cn/simple临时使用可以在使用pip的时候加参数-ihttps://pypi.tuna.tsinghua.edu.cn/simple例如:pipinstall-ihttps://

    2022年4月6日
    568

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号