用工具进行代码版本升级

在TensorFlow 2.x版本中，提供了一个升级TensorFlow 1.x版本代码的工具——tf_upgrade_v2。该工具可以非常方便地将TensorFlow 1.x版本中编写的代码移植到TensorFlow 2.x中。

tf_upgrade_v2 --infile "1.x代码文档"  -outfile "2.x代码文档"

tf_upgrade_v2工具支持单文档转换和多文档批量转换两种方式。

1、单个代码升级

tf_upgrade_v2 --infile foo_v1.py  --outfile foo_v2.py

 2、批量代码升级

tf_upgrade_v2 -intree foo_v1  -outtree foo_v2

 

这里的foo_v1指的是1.x代码文档目录，foo_v2指的是升级后的2.x代码文档存放目录。

需要注意的是tf_upgrade_v2工具并不是万能的，它只能实现基本的API升级。一般在转化完成之后还需要手动二次修改。