Browse Source

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

Merge pull request !3461 from Xiaoda/12-add-annotations-for-embeddinglookup
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
ea58b9be61
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ops/operations/array_ops.py

+ 1
- 1
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.


Loading…
Cancel
Save