該問題是在tfv2環境下跑了tfv1的代碼導致的,只需把v1的用法替換成v2留出的兼容用法即可:
用tf.compat.v1.nn.rnn_cell.RNNCell
代替tf.nn.rnn_cell.RNNCell
參考:https://tensorflow.google.cn/api_docs/python/tf/compat/v1/nn/rnn_cell/RNNCell?hl=en
也可以這樣,使2.0的環境能夠運行1.0的代碼:
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()