forked from mindspore-Ecosystem/mindspore
add output
This commit is contained in:
parent
3f7d5fc8c5
commit
5593d19c57
|
@ -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'):
|
||||
|
|
Loading…
Reference in New Issue