p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!21183 Rectify the Tensor shape under array_ops file. 

Merge pull request !21183 from dinglinhe/code_docs_dlh_ms_I3R3BX_9
This commit is contained in:
i-robot 2021-08-04 09:15:24 +00:00 committed by Gitee
parent b1b577b0ee 96d58ec523
commit 7b6822b9d4
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ops/operations/array_ops.py
View File

@ -739,6 +739,7 @@ class Unique(Primitive):
Inputs:
- **input_x** (Tensor) - The input tensor.
The shape is :math:`(N,*)` where :math:`*` means, any number of additional dimensions.
Outputs:
Tuple, containing Tensor objects `(y, idx), `y` is a tensor with the
@ -1369,6 +1370,7 @@ class OnesLike(Primitive):
Inputs:
- **input_x** (Tensor) - Input tensor.
The shape is :math:`(N,*)` where :math:`*` means, any number of additional dimensions.
Outputs:
Tensor, has the same shape and type as `input_x` but filled with ones.
@ -1401,6 +1403,7 @@ class ZerosLike(Primitive):
Inputs:
- **input_x** (Tensor) - Input tensor. The data type is int32, int64, float16 or float32.
The shape is :math:`(N,*)` where :math:`*` means, any number of additional dimensions.
Outputs:
Tensor, has the same shape and data type as `input_x` but filled with zeros.
@ -1716,6 +1719,7 @@ class Argmin(PrimitiveWithInfer):
Inputs:
- **input_x** (Tensor) - Input tensor.
The shape is :math:`(N,*)` where :math:`*` means, any number of additional dimensions.
Outputs:
Tensor, indices of the min value of input tensor across the axis.
@ -2658,6 +2662,7 @@ class Slice(PrimitiveWithInfer):
Inputs:
- **input_x** (Tensor): The target tensor.
The shape is :math:`(N,*)` where :math:`*` means, any number of additional dimensions.
- **begin** (Union[tuple, list]): The beginning of the slice. Only constant value(>=0) is allowed.
- **size** (Union[tuple, list]): The size of the slice. Only constant value is allowed.
@ -2733,6 +2738,7 @@ class ReverseV2(PrimitiveWithInfer):
Inputs:
- **input_x** (Tensor) - The target tensor. The data type is Number except float64.
The shape is :math:`(N,*)` where :math:`*` means, any number of additional dimensions.
Outputs:
Tensor, has the same shape and type as `input_x`.
@ -2795,7 +2801,7 @@ class Rint(PrimitiveWithInfer):
Inputs:
- **input_x** (Tensor) - The target tensor, which must be one of the following types:
float16, float32.
float16, float32. The shape is :math:`(N,*)` where :math:`*` means, any number of additional dimensions.
Outputs:
Tensor, has the same shape and type as `input_x`.