python - 如何在 Tensorflow 中实现堆叠式 RNN？

Question

我想在 GPU 上使用 Tensorflow1.13 实现 RNN。按照官方的建议，我编写了以下代码来获取一堆RNN单元

lstm = [tk.layers.CuDNNLSTM(128) for _ in range(2)]
cells = tk.layers.StackedRNNCells(lstm)

但是，我收到一条错误消息:

ValueError: ('All cells must have a state_size attribute. received cells:', [< tensorflow.python.keras.layers.cudnn_recurrent.CuDNNLSTM object at 0x13aa1c940>])

我该如何纠正它？

Answer 1

这可能是一个 Tensorflow 错误，我建议在 Github 上创建一个问题。但是，如果您想绕过该错误，可以使用:

import tensorflow as tf
import tensorflow.keras as tk

lstm = [tk.layers.CuDNNLSTM(128) for _ in range(2)]
stacked_cells = tf.nn.rnn_cell.MultiRNNCell(lstm)

这可以工作，但会给出一个您可以抑制的弃用警告。