使用pytorch中经常涉及到tensor和numpy数组之间的转换。下面就和小编一起来看下,torch.tensor和numpy.ndarray相互转换关系吧
工具/原料
1
python
2
pytorch
3
numpy
4
电脑
加载集成库
加载集成库首先需要加载集成库,才能使用库中定义的数据类型使用指令:【import 库名】
ndarray转换为tensor
1
step1 初始化ndarray在使用中,我们需要初始化ndarray,本例中使用直接赋值的方式使用指令:【numpy.array([data])】
2
step2 转换使用torch集成库中的转换函数可直接将ndarray进行转换为tensor,如图可以看到t已经成为了tensor使用指令:【torch.from_numpy(ndarray)】
3
step3 数据同步在使用时,我们仍需注意一个问题:经过变换后得到的tensor和ndarray数据上保持一致,ndarray修改了数据,tensor也会对应被修改数据
tensor转换为ndarray
1
step1 初始化tensor在使用中,我们需要初始化tensor,本例中使用直接赋值的方式使用指令:【torch.tensor([data])】
2
step2 转换使用torch集成库中的转换函数可直接将tensor进行转换为ndarray,如图可以看到a1已经成为了numpy.ndarray使用指令:【tensor.numpy()】
3
step3 数据同步在使用时,我们仍需注意一个问题:经过变换后得到的ndarray和原tensor数据上保持一致,tensor修改了数据,ndarray也会对应被修改数据
注意事项
使用时,一定要注意数据的同步问题,避免引入难以解决的bug
上一篇:从这11个方面来选安全的儿童推车
下一篇:哪些品牌的婴儿车好婴儿车采购要点