一、Pytorch简介
Pytorch是2017年Facebook人工智能研究院团队在Github上开源的一个深度学习框架,其历史可追溯到2002年诞生于纽约大学的Torch。自发布之后,Pytorch的热度便持续高涨,随着深度学习持续火热,Pytorch这一深度学习框架被运用地越发广泛。
二、为什么选择Pytorch
Pytorch是目前比较难得的简洁优雅且高效快速的框架。主要的优点有一下几个方面。
1.简洁
Pytorch的设计追求最少的封装,尽量避免重复造轮子。Pytorch的设计遵循tensor(张量,即高维数组)
variable(autograd)(变量,自动求导)
nn.Module(神经网络模块等)这三个由低到高的抽象层次,而且这三个层次联系紧密,可以同时进行修改和操作。设计的简洁的最大的好处就是代码易于理解。
2.速度
在很多评测中,Pytorch的速度表现胜过Tensorflow和Keras等框架。
3.易用
Pytorch的设计最符合人们的思维,它让用户尽可能专注于实现自己的想法,不需要考虑太多关于框架本身的束缚。
4.社区活跃
Pytorch提供了完整文档供用户学习,网上也有很多相关的教程供大家参考。也有相关的论坛去讨论使用Pytorch时遇到的问题,如Pytorch Forums等,同时也有很多的论文开始采用Pytorch作为论文实验工具,大家比较熟知的斯坦福大学的深度学习公开课,也在今年重新开课使用Pytorch作为开发框架,使用的人和学习的人越来越多