Tensor 和 NumPy 相互转换的实现
更新时间：2023年02月10日 10:10:52 作者：xzw96
本文主要介绍了Tensor 和 NumPy 相互转换的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
我们很容易用numpy()和from_numpy()将Tensor和NumPy中的数组相互转换。但是需要注意的一点是： 这两个函数所产生的Tensor和NumPy中的数组共享相同的内存（所以他们之间的转换很快），改变其中一个时另一个也会改变！
1. Tensor 转 NumPy
a = torch.ones(6) b = a.numpy() print(a, b) a += 1 print(a, b) b += 1 print(a, b)
tensor([1., 1., 1., 1., 1., 1.]) [1. 1. 1. 1. 1. 1.] tensor([2., 2., 2., 2., 2., 2.]) [2. 2. 2. 2. 2. 2.] tensor([3., 3., 3., 3., 3., 3.]) [3. 3. 3. 3. 3. 3.]
2. NumPy 数组转 Tensor
import numpy as np a = np.ones(7) b = torch.from_numpy(a) print(a, b) a += 1 print(a, b) b += 1 print(a, b)
[1. 1. 1. 1. 1. 1. 1.] tensor([1., 1., 1., 1., 1., 1., 1.], dtype=torch.float64) [2. 2. 2. 2. 2. 2. 2.] tensor([2., 2., 2., 2., 2., 2., 2.], dtype=torch.float64) [3. 3. 3. 3. 3. 3. 3.] tensor([3., 3., 3., 3., 3., 3., 3.], dtype=torch.float64)
3. torch.tensor() 将 NumPy 数组转换成 Tensor
直接用torch.tensor()将NumPy数组转换成Tensor，该方法总是会进行数据拷贝，返回的Tensor和原来的数据不再共享内存。
import numpy as np
a = np.ones((2,3))
c = torch.tensor(a)
a += 1
print('a:',a)
print('c:',c)
print(id(a)==id(c))a: [[2. 2. 2.]
[2. 2. 2.]]
c: tensor([[1., 1., 1.],
[1., 1., 1.]], dtype=torch.float64)
False
到此这篇关于Tensor 和 NumPy 相互转换的实现的文章就介绍到这了,更多相关Tensor 和 NumPy 相互转换内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要给大家介绍了因为Python升级导致yum、pip报错的解决方法，文中通过示例代码将解决的方法介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习下吧。2017-09-09
在Python中,format()函数是一种用于格式化字符串的方法主要介绍了Python中使用format函数的小结,本文就来介绍一下format()函数的使用示例,感兴趣的可以了解一下2023-08-08
进程彼此之间互相隔离，要实现进程间通信（IPC），multiprocessing模块主要通过队列方式，队列：队列类似于一条管道，元素先进先出，需要注意的一点是：队列都是在内存中操作,进程退出,队列清空,另外,队列也是一个阻塞的形态2022-12-12
这篇文章主要介绍了Vs Code中8个好用的python 扩展插件，帮助大家更好的利用vs code进行python开发，感兴趣的朋友可以了解下2020-10-10
我们经常在B站或其他视频网站上看到那种逐条读单词的视频,但他们的视频多多少少和我们的预期都不太一致,然而,网上很难找到和自己需求符合的视频,所以本文给大家介绍了使用Python制作读单词视频的实现,需要的朋友可以参考下2024-04-04
这篇文章主要为大家介绍了探索Python定时任务高效实现高效时间管理,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2024-01-01
这篇文章主要介绍了python 自定义装饰器,结合实例形式分析了Python自定义装饰器的功能、原理、用法及相关操作注意事项,需要的朋友可以参考下2019-07-07
本文介绍了Python中将字符串转换为列表的八种常用方法,包括split()方法、列表解析、正则表达式、str()函数、map()函数、re.split()、re.finditer()和逐字符遍历,感兴趣的可以了解一下2024-07-07
今天小编就为大家分享一篇在Python中构建增广矩阵的实现方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
django的分页器Paginator 从django中导入类
这篇文章主要介绍了django的分页器Paginator 从django中导入类,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-07-07
最新评论