Merge pull request #6956 from chengduoZH/feature/fix_embedding_example

Fix embedding example
del_some_in_makelist
chengduo 7 years ago committed by GitHub
commit d3c42f7d37
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -163,8 +163,9 @@ def embedding(input, size, is_sparse=False, param_attr=None, dtype='float32'):
Examples:
.. code-block:: python
dict_size = len(dataset.ids)
data = fluid.layers.data(name='ids', shape=[32, 32], dtype='float32')
fc = fluid.layers.embedding(input=data, size=16)
fc = fluid.layers.embedding(input=data, size=[dict_size, 16])
"""
helper = LayerHelper('embedding', **locals())

Loading…
Cancel
Save