
TensorFlow 2.0目前放出了Preview版本。目前想了解TensorFlow 2.0最快的方式就是查看官方API中的变化。本文列举一些从TF官方API中看出的TF 2.0的改进。
TensorFlow 2.0 Preview可以通过pip下载了:
- CPU: https://pypi.org/project/tf-nightly-2.0-preview/
- GPU: https://pypi.org/project/tf-nightly-gpu-2.0-preview/
API 文档也已开放:
- https://www.tensorflow.org/versions/r2.0/api_docs/python/tf
结合文档和一些新闻,TensorFlow 2.0 preview相对于TensorFlow 1.x在API上有下面一些重要的改变:
移除了tf.contrib
正如Google Group中所说,TensorFlow 2.0将不再维护tf.contrib模块。

移除了tf.layers
tf.layers是TensorFlow 1.x中被应用最广泛的package之一了,因为大部分的层都被封装在了tf.layers中。

考虑API统一的原因,TensorFlow统一将层放到了tf.keras.layers中。

移除了tf.train下的优化器
由于几乎所有的训练都要使用优化器,所以tf.train也是TensorFlow 1.x中高频使用的package,但在TensorFlow 2.0 Preview版API中,tf.train中的优化器都不见了。

与tf.layers类似,考虑到API的统一,现在优化器都放在了tf.keras.optimizers下面:

默认使用Eager Execution(动态图模式)
TensorFlow 2.0 Preview默认使用动态图模式,不需要手动开启了。
参考文档:
- https://www.tensorflow.org/versions/r2.0/api_docs/python/tf