提交 ea58b9be 编写于 作者: M mindspore-ci-bot 提交者: Gitee

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

Merge pull request !3461 from Xiaoda/12-add-annotations-for-embeddinglookup
......@@ -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.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册