p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!26704 Fix Embedding APi 

Merge pull request !26704 from huangxinjing/code_docs_fix_embedding
This commit is contained in:
i-robot 2021-11-24 01:59:15 +00:00 committed by Gitee
parent 87cbffc456 55efaabe2a
commit 5b4dfee77c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/parallel/nn/transformer.py
View File

@ -487,7 +487,7 @@ class VocabEmbedding(Cell):
- **output** (Tensor) - The embedding vector for the input with shape (batch_size,
seq_length, embedding_size).
- **weight** (Tensor) - The embedding table with shape (vocab_size, embedding_size).
- **embedding_table** (Tensor) - The embedding table with shape (vocab_size, embedding_size).
Raises:
ValueError: If the parallel_config.vocab_emb_dp is True, the vocab size is not a multiple of