本文来源自斯坦福大学计算机科学博士生 Awni Hannun 发表的文章,对比当前两个深度学习主流框架:TensorFlow 与 Pytorch 之间的差异。整体而言:
PyTorch 更适合于在研究中快速进行原型设计、业余爱好者和小型项目;而 TensorFlow 则更适合大规模的调度,尤其当考虑到跨平台和嵌入式调度操作时。细分对比如下 ——
准备时间 –> PyTorch
PyTorch 实际上是 NumPy 的替代,它支持 GPU,有着更高级的功能,可以用来构建和训练深度神经网络。如果你熟悉 NumPy、Python 和常见的深度学习概念(卷积层、递归层、SGD 等),那么[……]