Browse Source

update docment of KLDivLoss

pull/14763/head
liuhe 4 years ago
parent
commit
23a779e2fd
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ops/operations/nn_ops.py

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

@@ -4187,7 +4187,7 @@ class Adam(PrimitiveWithInfer):
TypeError: If `beta1_power`, `beta2_power1`, `lr`, `beta1`, `beta2`, `epsilon` or `gradient` is not a Tensor.

Supported Platforms:
``Ascend`` ``GPU``
``Ascend`` ``GPU`` ``CPU``

Examples:
>>> import numpy as np
@@ -4871,7 +4871,7 @@ class KLDivLoss(PrimitiveWithInfer):
L = \{l_1,\dots,l_N\}^\top, \quad
l_n = y_n \cdot (\log y_n - x_n)

Then,
Then,

.. math::
\ell(x, y) = \begin{cases}


Loading…
Cancel
Save