|
|
|
@@ -606,9 +606,9 @@ class BCELoss(_Loss): |
|
|
|
|
|
|
|
.. math:: |
|
|
|
\ell(x, y) = \begin{cases} |
|
|
|
L, & \text{if reduction} = \text{`none';}\\ |
|
|
|
\operatorname{mean}(L), & \text{if reduction} = \text{`mean';}\\ |
|
|
|
\operatorname{sum}(L), & \text{if reduction} = \text{`sum'.} |
|
|
|
L, & \text{if reduction} = \text{'none';}\\ |
|
|
|
\operatorname{mean}(L), & \text{if reduction} = \text{'mean';}\\ |
|
|
|
\operatorname{sum}(L), & \text{if reduction} = \text{'sum'.} |
|
|
|
\end{cases} |
|
|
|
|
|
|
|
Note: |
|
|
|
@@ -630,9 +630,9 @@ class BCELoss(_Loss): |
|
|
|
Otherwise, the output is a scalar. |
|
|
|
|
|
|
|
Raises: |
|
|
|
TypeError: If dtype of `inputs`, `labels` or `weight`(if given) is neither float16 not float32. |
|
|
|
TypeError: If dtype of `inputs`, `labels` or `weight` (if given) is neither float16 not float32. |
|
|
|
ValueError: If `reduction` is not one of 'none', 'mean', 'sum'. |
|
|
|
ValueError: If shape of `inputs` is not the same as `labels` or `weight`(if given). |
|
|
|
ValueError: If shape of `inputs` is not the same as `labels` or `weight` (if given). |
|
|
|
|
|
|
|
Supported Platforms: |
|
|
|
``Ascend`` ``GPU`` |
|
|
|
|