tf.keras.layers.GRUCell改为tf.keras.layers.LSTMCell出现IndexError: list index out of range

问题遇到的现象和发生背景

使用tensorflow构建RNN网络模型，把其中GRUCell改为LSTMCell

问题相关代码，请勿粘贴截图

    self.edge_update = tf.keras.layers.GRUCell(self.hparams.link_state_dim, name="edge_update")
    self.path_update = tf.keras.layers.GRUCell(self.hparams.path_state_dim, name="path_update")

把其中GRUCell改为LSTMCell

运行结果及报错内容

IndexError: list index out of range

我的解答思路和尝试过的方法

应该是这里出现了问题：
outputs, path_state = tf.nn.dynamic_rnn(self.path_update,
link_inputs,
sequence_length=lens,
initial_state = path_state,
dtype=tf.float32)

我想要达到的结果