Tensor和NumPy相互转换
2023-09-14 09:08:53 时间
Tensor 和 NumPy 相互转换常使用 numpy() 和 from_numpy() 。需要注意的是: 这两个函数所产生的 Tensor 和 NumPy 中的数组共享相同的内存(所以他们之间的转换很快),改变其中一个时另一个也会改变!
Tensor 转 Numpy 数组
a = torch.ones(3)
b = a.numpy()
print(a)
print(b)
tensor([1., 1., 1.])
[1. 1. 1.]
a += 1
print(a)
print(b)
a = a + 1
print(a)
print(b)
tensor([3., 3., 3.])
[3. 3. 3.]
tensor([4., 4., 4.])
[3. 3. 3.]
a = torch.ones(3)
b = a.numpy()
print(a)
print(b)
b += 1
print(a)
print(b)
b = b+1
print(a)
print(b)
tensor([1., 1., 1.])
[1. 1. 1.]
tensor([2., 2., 2.])
[2. 2. 2.]
tensor([2., 2., 2.])
[3. 3. 3.]
NumPy 数组转 Tensor
import numpy as np
a = np.ones(3)
b = torch.from_numpy(a)
print(a, b)
a += 1
print(a, b)
b += 1
print(a, b)
[1. 1. 1.] tensor([1., 1., 1.], dtype=torch.float64)
[2. 2. 2.] tensor([2., 2., 2.], dtype=torch.float64)
[3. 3. 3.] tensor([3., 3., 3.], dtype=torch.float64)
使用 torch.tensor() 将 NumPy 数组转换成 Tensor(不再共享内存)
c = torch.tensor(a) a += 1 print(a, c) [4. 4. 4.] tensor([3., 3., 3.], dtype=torch.float64)
相关文章
- LINK : fatal error LNK1123: 转换到 COFF 期间失败: 文件无效或损坏
- Java中List, Integer[], int[]的相互转换
- length、length()、size()区别 List与String相互转换
- numpy转换
- UTF-8, Unicode, GB2312格式串转换之C语言版
- RS-485收发的零延时转换电路
- 装箱转换
- php xml与json间的相互转换
- 布尔莎七参数坐标转换模型
- Atitit 网络协议概论 艾提拉著作 目录 1. 有的模型分七层,有的分四层。我觉得1 1.1. 三、链接层 确定了0和1的分组方式1 1.2. 网络层(ip mac转换层3 1.3. 传输
- CV:Image.open 和cv2.imread的简介、区别及PIL.Image格式/OpenCV格式相互转换代码实现之详细攻略
- 使用java进行文件编码转换
- python时间模块time,时间戳,结构化时间,字符串时间,相互转换,datetime
- FPGA-图像处理-色彩空间转换(RGB转YUV和YCBCR)
- 【多样化图像转换】2、You Only Need Adversarial Supervision for Semantic Image Synthesis(OASIS)代码