|
|
@@ -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 |
|
|
|