有 tf1 版本的代码,参考
如果你是 TensorFlow 1.X 的使用者,请参考 https://mp.weixin.qq.com/s?__biz=MzU1OTMyNDcxMQ==&mid=2247487599&idx=1&sn=13a53532ad1d2528f0ece4f33e3ae143&chksm=fc185b27cb6fd2313992f8f2644b0a10e8dd7724353ff5e93a97d121cd1c7f3a4d4fcbcb82e8&scene=21#wechat_redirect 以将代码转换为 TensorFlow 2 的 API。转换过程中有具体问题可另行提问。