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