替换为“来自tensorflow.contrib导入层”

时间:2018-11-28 07:04:32

标签: tensorflow

如何用新的核心功能替换from tensorflow.contrib import layers。我想将TF 1.4代码移至1.12,以准备TF 2.0。

1 个答案:

答案 0 :(得分:1)

tf.contrib.layers相对应的核心功能在tf.layers中。 this question中讨论了一些差异。但是,这不会为您准备TF 2.0。

如果您的目标是为TF 2.0准备代码,请考虑将tf.contrib完全删除(从TF拆分或集成到TF中),并将tf.layers也删除,并且将级API将位于tf.keras下。因此,为了最好地为TF 2.0做准备,您应该开始使用tf.keras.layers

这是blog post,涉及TF 2.0可能会遇到的一些实际差异。