提交 9efd85cd 编写于 作者: Y Yan Xu 提交者: Cheerego

fix embedding layer example (#732)

上级 4d4e04be
...@@ -3721,8 +3721,8 @@ embedding ...@@ -3721,8 +3721,8 @@ embedding
.. code-block:: python .. code-block:: python
dict_size = len(dataset.ids) dict_size = len(dataset.ids)
data = fluid.layers.data(name='ids', shape=[32, 32], dtype='float32') data = fluid.layers.data(name='ids', shape=[1], dtype='int64')
fc = fluid.layers.embedding(input=data, size=[dict_size, 16]) fc = fluid.layers.embedding(input=data, size=[dict_size, 16], dtype='float32')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册