多语言展示
当前在线:1823今日阅读:27今日分享:41

Pytorch torch.tensor和numpy.ndarray相互转换

使用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

推荐信息