Tensor 和 NumPy 相互转换
2023-06-13 09:14:29 时间
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
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/180185.html原文链接:https://javaforall.cn
相关文章
- pycharm不能输入中文怎么办_电脑无法转换中文输入法
- 借助栈来实现单链表的逆置运算_中缀后缀表达式互相转换
- pytorch: tensor与numpy之间的转换[通俗易懂]
- Pytorch中tensor和numpy互相转换[通俗易懂]
- verilog vhdl混合_vhdl转换为verilog
- 【数据挖掘】聚类 Cluster 矩阵转换 数据矩阵 -> 相似度矩阵 ( 二元变量简介 | 二元变量可能性表 | 对称二元变量 | 简单匹配系数 | 非对称二元变量 | Jaccard 系数 )
- fastjson对json字符串JSONObject和JSONArray互相转换操作示例详解编程语言
- int转换char的正确姿势详解编程语言
- MySQL NULL值转换为0的实现(mysqlnull转0)
- MySQL转换为字符串的简单方法(mysql转换成字符串)
- MySQL中实现日期转换的方法(mysqldate转换)
- 利用Oracle实现UTC时间转换(oracleutc转换)
- 转换快速掌握Linux下视频格式转换技巧(linux视频格式)
- Linux环境下如何转换二进制文件(linux转换二进制文件)
- MySQL转换为字符串的技巧(mysql转换为字符串)
- Oracle数据库从内存转换到数字(oracle内存转数字)
- 深入浅出从MySQL到Bit的转换方法(bit转换mysql)
- 解决DBF数据转换至Oracle的解决方案(dbf转换oracle)
- Oracle数据库如何将时间转换为季度(oracle中时间转季度)
- 将Redis缓存转换为数据库的进阶方法(redis缓存转数据库)
- c#转换全角半角方法示例