Python: 数据类型转换总结(list-np.array-torch.tensor)

Python: 数据类型转换总结

总结记录在深度学习中经常用到的list,np,array,和torch.tensor数据类型的转换过程。

1.listnp.arraytorch.tensor数据类型转换

  • 定义三种类型变量
  • 简单list类型:l = [1, 2, 3, 4, 5]
  • np.array类型:a = np.array([1, 2, 3, 4, 5])
  • torch.tensor类型:t = torch.tensor([1, 2, 3, 4, 5])

1.1 listnp.array之间转换

1.1.1 listtonp.array
复制代码
l2a = np.array(l)
1.1.2 np.arraytolist
复制代码
a2l = a.tolist()

1.2 listtorch.tensor之间转换

1.2.1 listtotorch.tensor
复制代码
l2t = torch.tensor(l)
1.2.2 torch.tensortolist
复制代码
t2l = t.tolist()

1.3 np.arraytorch.tensor之间转换

1.3.1 np.arraytotorch.tensor
复制代码
a2t = torch.from_numpy(a)
1.3.2 torch.tensortonp.array之间转换
复制代码
t2a = t.numpy()

1.4. list[torch.tensor]tolist之间转换

复制代码
d = [torch.tensor(1), torch.tensor(2), torch.tensor(3), torch.tensor(4)]
d2l = [tensor.tolist() for tensor in d]
相关推荐
ouliten5 分钟前
cuda编程笔记(29)-- CUDA Graph
笔记·深度学习·cuda
星期天要睡觉8 分钟前
深度学习——基于 ResNet18 的图像分类训练
pytorch·python·机器学习
林炳然9 分钟前
Python-Basic Day-1 基本元素(数字、字符串)
python
weixin_3077791312 分钟前
在Linux服务器上使用Jenkins和Poetry实现Python项目自动化
linux·开发语言·python·自动化·jenkins
润 下12 分钟前
C语言——深入解析C语言指针:从基础到实践从入门到精通(四)
c语言·开发语言·人工智能·经验分享·笔记·程序人生·其他
今天没有盐14 分钟前
内置基础类型之布尔值类型(bool)与时间与日期类型
python·编程语言
Empty_77716 分钟前
Python编程之常用模块
开发语言·网络·python
hzp66618 分钟前
Magnus:面向大规模机器学习工作负载的综合数据管理方法
人工智能·深度学习·机器学习·大模型·llm·数据湖·大数据存储
小火柴12320 分钟前
利用R绘制箱线图
开发语言·r语言
m0_6786933326 分钟前
深度学习笔记39-CGAN|生成手势图像 | 可控制生成(Pytorch)
深度学习·学习·生成对抗网络