BCELoss如何处理log(0)?

时间:2019-06-17 19:44:01

标签: pytorch

Pytorch中的BCELoss()如何处理log(0)?

crit1 = nn.BCELoss()
output = torch.tensor([1., 0., 1., 0.])
target = torch.tensor([1., 1., 1., 0.])
print(crit1(output, target))

这将输出tensor(6.9078),而设置reduction='none'将输出tensor([-0.0000, 27.6310, -0.0000, -0.0000])。如果有人可以遍历计算来找到这些数字,将不胜感激。

0 个答案:

没有答案