m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

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

Browse Source 
Merge pull request !3461 from Xiaoda/12-add-annotations-for-embeddinglookup
pull/3461/MERGE
mindspore-ci-bot 5 years ago 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

2
mindspore/ops/operations/array_ops.py
Unescape View File

@ -3424,7 +3424,7 @@ class EmbeddingLookup(PrimitiveWithInfer):
Inputs: Inputs:
- **input_params** (Tensor) - The shape of tensor is :math:`(x_1, x_2, ..., x_R)`. - **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)`. - **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`, 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. and the exceeding part will be filled with 0 in the output.

Write Preview
Loading…
Cancel
Save