add output

This commit is contained in:
wanyiming 2020-10-29 20:04:54 +08:00
parent 3f7d5fc8c5
commit 5593d19c57
1 changed files with 1 additions and 0 deletions

View File

@ -487,6 +487,7 @@ class BCELoss(_Loss):
>>> inputs = Tensor(np.array([[0.1, 0.2, 0.3], [0.5, 0.7, 0.9]]), mindspore.float32)
>>> labels = Tensor(np.array([[0, 1, 0], [0, 0, 1]]), mindspore.float32)
>>> loss(inputs, labels)
1.8952923
"""
def __init__(self, weight=None, reduction='none'):