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

!3461 [EmbeddingLookup] Add the dimension constraint on the Parameter 

Merge pull request !3461 from Xiaoda/12-add-annotations-for-embeddinglookup
This commit is contained in:
mindspore-ci-bot 2020-07-27 10:17:48 +08:00 committed by Gitee
parent 0708b18165 a0a08b1d02
commit ea58b9be61
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/ops/operations/array_ops.py
View File

@ -3424,7 +3424,7 @@ class EmbeddingLookup(PrimitiveWithInfer):
Inputs:
- **input_params** (Tensor) - The shape of tensor is :math:`(x_1, x_2, ..., x_R)`.
The Tensor slice, instead of the entire Tensor.
This represents a Tensor slice, instead of the entire Tensor. Currently, the dimension is restricted to be 2.
- **input_indices** (Tensor) - The shape of tensor is :math:`(y_1, y_2, ..., y_S)`.
Specifies the indices of elements of the original Tensor. Values can be out of range of `input_params`,
and the exceeding part will be filled with 0 in the output.