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])
。如果有人可以遍历计算来找到这些数字,将不胜感激。