From bf5a814672c795b2565a4f9cf35e0dcb9dd9ffc2 Mon Sep 17 00:00:00 2001 From: dinglinhe Date: Fri, 30 Jul 2021 09:27:19 +0800 Subject: [PATCH] add support data type to ArgMax and fix some error about Floor operator --- mindspore/ops/operations/array_ops.py | 7 ++++++- mindspore/ops/operations/math_ops.py | 4 ++-- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py index 81282df727d..92ad2b1759b 100755 --- a/mindspore/ops/operations/array_ops.py +++ b/mindspore/ops/operations/array_ops.py @@ -1654,7 +1654,12 @@ class Argmax(PrimitiveWithInfer): Default: `mindspore.dtype.int32`. Inputs: - - **input_x** (Tensor) - Input tensor. + - **input_x** (Tensor) - Input tensor. :math:`(N,*)` where :math:`*` means, any number of additional dimensions. + Support data type list as follows: + + - Ascend: Float16, Float32. + - GPU: Float16, Float32. + - CPU: Float16, Float32, Float64. Outputs: Tensor, indices of the max value of input tensor across the axis. diff --git a/mindspore/ops/operations/math_ops.py b/mindspore/ops/operations/math_ops.py index 26422b50062..7d4dd49cee4 100644 --- a/mindspore/ops/operations/math_ops.py +++ b/mindspore/ops/operations/math_ops.py @@ -2634,7 +2634,7 @@ class FloorDiv(_MathBinaryOp): out_{i} = \\text{floor}( \\frac{x_i}{y_i}) - where the :math:`floor` indicates the operator that converts the input data into the floor data type. + where the :math:`floor` indicates the Floor operator, for more details, please refer to the Floor operator. Inputs: - **x** (Union[Tensor, Number, bool]) - The first input is a number or @@ -2841,7 +2841,7 @@ class FloorMod(_MathBinaryOp): out_{i} =\text{floor}(x_{i} // y_{i}) - where the :math:`floor` indicates the operator that converts the input data into the floor data type. + where the :math:`floor` indicates the Floor operator, for more details, please refer to the Floor operator. Inputs: - **x** (Union[Tensor, Number, bool]) - The first input is a number or