There was an error while loading. Please reload this page.
1 parent ac9245a commit 5b171adCopy full SHA for 5b171ad
torch/nn/modules/loss.py
@@ -191,8 +191,7 @@ class BCELoss(_WeightedLoss):
191
..math:: loss(o, t) = - 1/n \sum_i weights[i] * (t[i] * log(o[i]) + (1 - t[i]) * log(1 - o[i]))
192
193
This is used for measuring the error of a reconstruction in for example
194
- an auto-encoder. Note that the targets `t[i]` should be numbers between 0 and 1,
195
- for instance, the output of an `nn.Sigmoid` layer.
+ an auto-encoder. Note that the targets `t[i]` should be numbers between 0 and 1.
196
197
By default, the losses are averaged for each minibatch over observations
198
*as well as* over dimensions. However, if the field `sizeAverage` is set
0 commit comments