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)
全栈程序员-站长的头像全栈程序员-站长


相关推荐

  • 推荐一个通用的免费传真网站「建议收藏」

    推荐一个通用的免费传真网站「建议收藏」      当我们没有传真机,而又想发送传真时。我们可以试试Myfax提供的免费在线发传真,发送方无需安装传真机,只要对方有传真机接收就可以了,操作非常简单。       首先登陆免费在线放传真的网站:http://www.myfax.com/free/,然后在打开的网站页面中,在TO下的Country框中,网站根据IP地址自动显示用户所处于的地理位置。单击右侧的下拉按钮,选择接收传真的国

    2022年6月28日
    65
  • cubieboard boot过程

    cubieboard boot过程A10的启动过程大概可分为5步:BootRom,SPL,Uboot,Kernel,RootFileSystem。本文只关注镜像的加载过程,分析RootRom->SPL->Uboot的启动流程。系统上电后,ARM处理器在复位时从地址0x000000开始执行指令,把板上ROM或Flash映射到这一地址。A10将启动设备选择程序固化在CPU内部的一个32KBROM中,默认的启动时序为SD

    2022年7月22日
    11
  • eXtremeDB HA synchronization procedure

    eXtremeDB HA synchronization procedureeXtremeDBHAsynchronizationprocedure

    2022年8月31日
    3
  • 怎么新建pytest的ini文件_python的配置文件ini

    怎么新建pytest的ini文件_python的配置文件ini前言pytest配置文件可以改变pytest的运行方式,它是一个固定的文件pytest.ini文件,读取配置信息,按指定的方式去运行查看pytest.ini的配置选项pytest-h找到以下

    2022年7月28日
    11
  • 跟我一起写 Makefile(二)

    跟我一起写 Makefile(二)三、make是如何工作的在默认的方式下,也就是我们只输入make命令。那么,   1、make会在当前目录下找名字叫“Makefile”或“makefile”的文件。   2、如果找到,它会找文件中的第一个目标文件(target),在上面的例子中,他会找到“edit”这个文件,并把这个文件作为最终的目标文件。   3、如果edit文件不存在,或是edit所依赖的后面的.o文件的文

    2022年5月4日
    41
  • Qt for android触摸手势事件QGestureEvent「建议收藏」

    Qt for android触摸手势事件QGestureEvent「建议收藏」在触摸设备上可以使用Qt的手势事件要激活手势事件,需要执行以下操作:第一步,为QWidget控件注册手势事件QListgestures;gesturesgesturesgesturesgesturesgesturesMainWindoww;w.grabGestures(gestur

    2025年9月4日
    7

发表回复

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

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