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

Merge pull request !3461 from Xiaoda/12-add-annotations-for-embeddinglookup
pull/3461/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit ea58b9be61

@ -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.

Loading…
Cancel
Save