lab04: states for LSTM

labs/OATD_LR4_metod.ipynb

@@ -918,7 +918,7 @@
     "    if states is None:\n",
     "        states = self.gru.get_initial_state(x)\n",
     "\n",
-    "    x, states = self.gru(x, initial_state=states, training=training)\n",
+    "    x, *states = self.gru(x, initial_state=states, training=training)\n",
     "    x = self.dense(x, training=training)\n",
     "\n",
     "    if return_state:\n",