!5414 fix bugs of op ReduceMean and Softsign

Merge pull request !5414 from lihongkang/lhk_master
pull/5414/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit c7b3c9ef50

@ -319,7 +319,7 @@ class ReduceMean(_Reduce):
Tensor, has the same dtype as the 'input_x'. Tensor, has the same dtype as the 'input_x'.
- If axis is (), and keep_dims is false, - If axis is (), and keep_dims is false,
the output is a 0-D tensor representing the sum of all elements in the input tensor. the output is a 0-D tensor representing the mean of all elements in the input tensor.
- If axis is int, set as 2, and keep_dims is false, - If axis is int, set as 2, and keep_dims is false,
the shape of output is :math:`(x_1, x_3, ..., x_R)`. the shape of output is :math:`(x_1, x_3, ..., x_R)`.
- If axis is tuple(int), set as (2, 3), and keep_dims is false, - If axis is tuple(int), set as (2, 3), and keep_dims is false,

@ -233,7 +233,7 @@ class Softsign(PrimitiveWithInfer):
The function is shown as follows: The function is shown as follows:
.. math:: .. math::
\text{output} = \frac{\text{input_x}}{1 + \abs{\text{input_x}}}, \text{output} = \frac{\text{input_x}}{1 + \left| \text{input_x} \right|},
Inputs: Inputs:
- **input_x** (Tensor) - The input tensor whose data type should be float16 or float32. - **input_x** (Tensor) - The input tensor whose data type should be float16 or float32.

Loading…
Cancel
Save