基于TensorFlow和PyTorch的深度学习框架对比分析

• 作者: 梁炜萱 肖祖环 黄玉萍
• 来源期刊: 现代信息科技
• 学科: 工学
• 关键词: 深度学习 TensorFlow PyTorch 适用场景
• 年，卷（期）: 2020,（4）
• 所属期刊栏目: 计算机技术
• 页码范围: 80-82,87
• 页数: 4页
• 分类号: TP181|TP391.41
• 字数: 3243字
• 语种: 中文

摘要

深度学习框架是实现机器学习的关键工具,合适的深度学习框架可以达到事半功倍的效果.为助力研究者选择合适的框架,在回顾近十种常见框架的基础上,聚焦当前受众最广、热度最高的两种深度学习框架TensorFlow和PyTorch,从历程、现状、机制、训练模式、可视化、工业部署等角度对两者进行比对分析,并归类对应适用场景的建议,为框架选择提供思路参考.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	黄玉萍		6	1	1.0	1.0
2	梁炜萱		1	0	0.0	0.0
3	肖祖环		1	0	0.0	0.0