Tensor, array, list之间的互相转化

一、数据定义

1、Tensor(torch、mindspore等框架下)

 张量是在深度学习框架中的一个数据结构,个人理解就是这个框架的输入数据必须是tensor格式,比如一张图片进来,需要转化成tensor格式输入到网络中,再在框架进行一系列的操作,等模型训练完了,用不到这个框架了,可以把这个tensor取出来,转换成别的需要进一步操作的数据类型(例如array,list等)

2、array(numpy)

数组结构是由不同维度的list转换来的,用array的原因主要在于有更多的矩阵操作,数据使用起来更方便,比如转置、矩阵相乘、reshape等等

二、互相转换

1、array转list

import numpy as np
a = np.array(12, np.float32)
list = a.tolist()

2、list转array

import numpy as np
a = list()
array = np.array(a)

3、array转Tensor

import numpy as np
a = np.array(1, np.float32)
tensor = torch.from_numpy(a)

4、Tensor转array(一般用到的就是框架生成的tensor取出来进行操作)

​
array = tensor.numpy()

​

5、Tensor转list

该转换不能一部完成,通常需要先把tensor转array,再把array转list

list = tensor.numpy().tolist()

6、list转Tensor

tensor=torch.Tensor(list)

参考原博客:

(13条消息) tensor转换为list_【串讲总结】array, list, tensor,Dataframe,Series之间互相转换总结…_weixin_39533174的博客-CSDN博客

物联沃分享整理
物联沃-IOTWORD物联网 » Tensor, array, list之间的互相转化

发表评论