|
@@ -99,13 +99,13 @@
|
|
|
"\n",
|
|
|
" # Prepare data shape to match `bidirectional_rnn` function requirements\n",
|
|
|
" # Current data input shape: (batch_size, n_steps, n_input)\n",
|
|
|
- " # Required shape: 'n_steps' tensors list of shape (batch_size, n_hidden)\n",
|
|
|
+ " # Required shape: 'n_steps' tensors list of shape (batch_size, n_input)\n",
|
|
|
" \n",
|
|
|
" # Permuting batch_size and n_steps\n",
|
|
|
" x = tf.transpose(x, [1, 0, 2])\n",
|
|
|
" # Reshape to (n_steps*batch_size, n_input)\n",
|
|
|
" x = tf.reshape(x, [-1, n_input])\n",
|
|
|
- " # Split to get a list of 'n_steps' tensors of shape (batch_size, n_hidden)\n",
|
|
|
+ " # Split to get a list of 'n_steps' tensors of shape (batch_size, n_input)\n",
|
|
|
" x = tf.split(0, n_steps, x)\n",
|
|
|
"\n",
|
|
|
" # Define lstm cells with tensorflow\n",
|