Python 带有元组输入的自定义TensorFlow RNN单元

我试图在TensorFlow中创建一个接受元组作为输入的自定义RNN单元，但遇到了一个问题，即父类

BasicLSTMCell

要求输入是二维的：

# Inputs must be 2-dimensional.
self.input_spec = base_layer.InputSpec(ndim=2)

---

我怎样才能绕过这个限制？我无法添加逻辑来处理

call（）

方法中的元组，因为执行从未到达该方法-维度检查会引发错误。

我实际上也发现了这个问题。tensorflow平台中有一个bug。您可以通过更改recurrent.py文件中的get_step_input_shape函数来解决此问题。只需将[0]添加到此行的末尾：nest.map\u structure（get\u input\u spec，input\u shape））