Tensorflow tf.compat.v1的替代方案

我试图将我的tf1代码迁移到tf2中

为此，我将以下功能更改如下：

tf.loss.softmax\u cross\u entropy（）

进入

tf.compat.v1.loss.softmax\u cross\u entropy（）

tf.train.MomentumOptimizer（）

进入

tf.compat.v1.train.MomentumOptimizer（）

tf.train.get\u或\u create\u global\u step（）

进入

tf.compat.v1.train.get\u或\u create\u global\u step（）

但是，我想直接在tf2中重写代码，而不使用

tf.compat.v1

---

怎么样？

要将TF1代码转换为TF2代码，可以通过在TF2中使用tf.keras api轻松完成。例如，您提供的功能

tf.loss.softmax\u cross\u entropy（）

to

tf.keras.loss.CategoricalCrossentropy（from\u logits=True）

from_logits

这里的参数指定y_pred是否应该是logits张量

tf.train.MomentumOptimizer（）

to

tf.keras.optimizers.SGD（momentum=…）

通过提供

动量

参数

至于最后一个函数，它仍在TF2.0文档中使用，因此2.0中可能没有等效版本

您可以从TF2.0网站查看本指南，该网站将为您提供一个很好的参考