具有元组输入的自定义 TensorFlow RNN 单元答案

【问题标题】：Custom TensorFlow RNN Cell with Tuple Input具有元组输入的自定义 TensorFlow RNN 单元
【发布时间】：2017-11-27 21:32:22
【问题描述】：

我正在尝试在 TensorFlow 中创建一个接受元组作为输入的自定义 RNN 单元，但我遇到了父类 BasicLSTMCell 要求输入是二维的问题：

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

我怎样才能绕过这个限制？我无法在 call() 方法中添加处理元组的逻辑，因为执行永远不会到达该方法 - 维度检查会引发错误。

【问题讨论】：

【解决方案1】：

其实我也发现了这个问题。 tensorflow平台有一个bug。您可以通过更改recurrent.py 文件中的get_step_input_shape 函数来解决。只需在此行末尾添加 [0]：nest.map_structure(get_input_spec, input_shape))

【讨论】：