Skip to content

GitLab

"docs/git@developer.sourcefind.cn:norm/vllm.git" did not exist on "c81dddb45c71e630b907f9d84686ecd73b4105c7"
Commit 6a72aec2 authored by Mark Daoust's avatar Mark Daoust
Browse files

Created using Colaboratory

parent a17440f7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
samples/core/get_started/eager.ipynb
View file @ 6a72aec2
...@@ -411,8 +411,7 @@ ...@@ -411,8 +411,7 @@
"cell_type": "code", "cell_type": "code",
"source": [ "source": [
"def pack_features_vector(features,labels):\n", "def pack_features_vector(features,labels):\n",
" values = [value for value in features.values()]\n", " features = tf.stack(list(features.values()), axis=1)\n",
" features = tf.stack(values, axis=1)\n",
" return features, labels\n", " return features, labels\n",
" \n", " \n",
"train_dataset = train_dataset.map(pack_features_vector)" "train_dataset = train_dataset.map(pack_features_vector)"
...@@ -1071,4 +1070,4 @@ ...@@ -1071,4 +1070,4 @@
] ]
} }
] ]
} }
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment