Created using Colaboratory

6a72aec2 · Mark Daoust · a17440f7 · 6a72aec2
Commit 6a72aec2 authored May 29, 2018 by Mark Daoust
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

samples/core/get_started/eager.ipynb samples/core/get_started/eager.ipynb +2 -3

No files found.
--- a/samples/core/get_started/eager.ipynb
+++ b/samples/core/get_started/eager.ipynb
@@ -411,8 +411,7 @@
      "cell_type": "code",
      "source": [
        "def pack_features_vector(features,labels):\n",
-        "  values = [value for value in features.values()]\n",
-        "  features = tf.stack(values, axis=1)\n",
+        "  features = tf.stack(list(features.values()), axis=1)\n",
        "  return features, labels\n",
        "  \n",
        "train_dataset = train_dataset.map(pack_features_vector)"