From 36f54d0573b5b4813b7179b3c88e13c330a9ad9f Mon Sep 17 00:00:00 2001 From: zhouyuanshen Date: Fri, 11 Dec 2020 16:39:50 +0800 Subject: [PATCH] add index value range lint for gatherd in front-end api --- 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 676ed79bf2..91ceca2b21 100644 --- a/mindspore/ops/operations/array_ops.py +++ b/mindspore/ops/operations/array_ops.py @@ -4584,7 +4584,7 @@ class GatherD(PrimitiveWithInfer): - **x** (Tensor) - The source tensor. - **dim** (int) - The axis along which to index. It must be int32. Only constant value is allowed. - **index** (Tensor) - The indices of elements to gather. It can be one of the following data types: - int32, int64. + int32, int64. The value range of each index element is [-x_rank[dim], x_rank[dim]). Outputs: Tensor, the shape of tensor is :math:`(z_1, z_2, ..., z_N)`.