Browse Source

add index value range lint for gatherd in front-end api

tags/v1.1.0
zhouyuanshen 5 years ago
parent
commit
36f54d0573
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ops/operations/array_ops.py

+ 1
- 1
mindspore/ops/operations/array_ops.py View File

@@ -4584,7 +4584,7 @@ class GatherD(PrimitiveWithInfer):
- **x** (Tensor) - The source tensor. - **x** (Tensor) - The source tensor.
- **dim** (int) - The axis along which to index. It must be int32. Only constant value is allowed. - **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: - **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: Outputs:
Tensor, the shape of tensor is :math:`(z_1, z_2, ..., z_N)`. Tensor, the shape of tensor is :math:`(z_1, z_2, ..., z_N)`.


Loading…
Cancel
Save