p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!31558 optimizes the documentation of english API of BinaryCrossEntropy 

Merge pull request !31558 from zhangfanghe/code_docs_BinaryCrossEntropy_319
This commit is contained in:
i-robot 2022-03-21 07:48:23 +00:00 committed by Gitee
parent 3b4ee09875 fa0af98848
commit aa2c1c590d
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/python/mindspore/ops/operations/nn_ops.py
View File

@ -5155,8 +5155,8 @@ class BinaryCrossEntropy(Primitive):
And it must have the same shape and data type as `logits`. Default: None.
Outputs:
Tensor or Scalar, if `reduction` is 'none', then output is a tensor and has the same shape as `logits`.
Otherwise, the output is a scalar.
Tensor, has the same dtype as `logits`. if `reduction` is 'none', then it has the same shape as `logits`.
Otherwise, it is a scalar Tensor.
Raises:
TypeError: If dtype of `logits`, `labels` or `weight` (if given) is neither float16 nor float32.