综述

名称 发布时间 阵营 源码 star fork
TensorFlow Nov 1, 2015 Google、Yahoo! tensorflow/tensorflow 62,648 30,331
Caffe Sep 8, 2013 BVLC BVLC/caffe 18,869 11,586
PyTorch Feb 2, 2017 Facebook pytorch/pytorch 5,718 1,071
Keras Mar 22, 2015 fchollet/keras 17,273 6,168
Theano Jan 6, 2008 Theano/Theano 6,583 2,191
MXNet Apr 26, 2015 Amazon dmlc/mxnet 10,280 3,847
CNTK Jul 27, 2014 Microsoft Microsoft/CNTK 11,661 2,953

深度学习: 深度学习框架 横向对比

Tensorflow

  • Python API 编写,通过 C/C++引擎 加速;

  • 使用 数据流图 生成 深度学习 中 最常见的 基本单元

  • 嵌套关系 (从外到内):

    1. Graph (每次只能在一个 指定的默认graph 中进行操作)
    2. Session (会去遍历那些具有依赖关系的op)
    3. Operation (节点)

[1] TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

相关文章: