|
|
|
@@ -4580,7 +4580,8 @@ class EmbeddingLookup(PrimitiveWithInfer): |
|
|
|
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. |
|
|
|
and the exceeding part will be filled with 0 in the output. Values does not support negative and the result |
|
|
|
is undefined if values are negative. |
|
|
|
- **offset** (int) - Specifies the offset value of this `input_params` slice. Thus the real indices |
|
|
|
are equal to `input_indices` minus `offset`. |
|
|
|
|
|
|
|
|