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