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博客