Browse Source

!2559 Fix some mistakes of ApplyCenteredRMSProp, aSinh and Sinh vm ops

Merge pull request !2559 from liuwenhao/rms
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
af2ebca023
2 changed files with 3 additions and 3 deletions
  1. +2
    -2
      mindspore/ops/operations/math_ops.py
  2. +1
    -1
      mindspore/ops/operations/nn_ops.py

+ 2
- 2
mindspore/ops/operations/math_ops.py View File

@@ -1888,7 +1888,7 @@ class Cosh(PrimitiveWithInfer):


class Asinh(PrimitiveWithInfer): class Asinh(PrimitiveWithInfer):
""" """
Compute inverse hyperbolic cosine of x element-wise.
Compute inverse hyperbolic sine of x element-wise.


Inputs: Inputs:
- **input_x** (Tensor) - The shape of tensor is :math:`(x_1, x_2, ..., x_R)`. - **input_x** (Tensor) - The shape of tensor is :math:`(x_1, x_2, ..., x_R)`.
@@ -2645,7 +2645,7 @@ class Sin(PrimitiveWithInfer):


class Asin(PrimitiveWithInfer): class Asin(PrimitiveWithInfer):
""" """
Computes arccosine of input element-wise.
Computes arcsine of input element-wise.


Inputs: Inputs:
- **input_x** (Tensor) - The shape of tensor is :math:`(x_1, x_2, ..., x_R)`. - **input_x** (Tensor) - The shape of tensor is :math:`(x_1, x_2, ..., x_R)`.


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

@@ -1905,7 +1905,7 @@ class ApplyCenteredRMSProp(PrimitiveWithInfer):
>>> mean_grad = Tensor(np.arange(12).astype(np.float32).reshape(2, 3, 2), mindspore.float32) >>> mean_grad = Tensor(np.arange(12).astype(np.float32).reshape(2, 3, 2), mindspore.float32)
>>> mean_square = Tensor(np.arange(-8, 4).astype(np.float32).reshape(2, 3, 2), mindspore.float32) >>> mean_square = Tensor(np.arange(-8, 4).astype(np.float32).reshape(2, 3, 2), mindspore.float32)
>>> moment = Tensor(np.arange(12).astype(np.float32).reshape(2, 3, 2), mindspore.float32) >>> moment = Tensor(np.arange(12).astype(np.float32).reshape(2, 3, 2), mindspore.float32)
>>> grad = Tensor(np.arange(12).astype(np.float32).rehspae(2, 3, 2), mindspore.float32)
>>> grad = Tensor(np.arange(12).astype(np.float32).reshape(2, 3, 2), mindspore.float32)
>>> learning_rate = Tensor(0.9, mindspore.float32) >>> learning_rate = Tensor(0.9, mindspore.float32)
>>> decay = 0.0 >>> decay = 0.0
>>> momentum = 1e-10 >>> momentum = 1e-10


Loading…
Cancel
Save