From a0a08b1d023b32704963e61534971e57d2407f38 Mon Sep 17 00:00:00 2001 From: Xiaoda Zhang Date: Sat, 25 Jul 2020 10:07:12 +0800 Subject: [PATCH] add dimension constraints on embeddinglookup --- mindspore/ops/operations/array_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py index 18b38f543bc..56ae84842b6 100644 --- a/mindspore/ops/operations/array_ops.py +++ b/mindspore/ops/operations/array_ops.py @@ -3423,7 +3423,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.