python - seq2seq - Entradas no una secuencia con RNNs y TensorFlow

tensorflow seq2seq

Question

Tengo un código lstm muy básico con tensorflow y python, donde mi código es

output = tf.nn.rnn(tf.nn.rnn_cell.BasicLSTMCell(10), input_flattened, initial_state=tf.placeholder("float", [None, 20]))

donde mi entrada aplanada tiene forma [?, 5, 22501]

Estoy obteniendo el error TypeError: inputs must be a sequence en el parámetro de state del lstm, y estoy arrancándome el pelo tratando de descubrir por qué me está dando este error. Cualquier ayuda sería muy apreciada.

Answer 1

Creo que cuando usas la función tf.nn.rnn está esperando una lista de tensores y no solo un tensor. Deberías desempaquetar la entrada en la dirección del tiempo para que sea una lista de tensores de forma [?, 22501]. También podría usar tf.nn.dynamic_rnn, que creo que puede manejar este desempaquetado para usted.