pytorch bceloss测试

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/jacke121/article/details/82809587

torch bceloss测试

x与y两个值差异越大，loss越大，

a与b都越接近1，loss越小

a=torch.FloatTensor([0.9,1])
b=torch.FloatTensor((0.9,1))

print(loss_fn(a,b))

import torch

a=torch.FloatTensor((0.9,0.1))
b=torch.FloatTensor((0.1,0.9))

loss_fn = torch.nn.BCELoss(reduce=False, size_average=False)

print(loss_fn(a,b))

a=torch.FloatTensor((0.2,0.9))
b=torch.FloatTensor((0.8,0.1))

print(loss_fn(a,b))

a=torch.FloatTensor((0.7,0.3))
b=torch.FloatTensor((0.3,0.7))

print(loss_fn(a,b))

a=torch.FloatTensor((0.6,0.1))
b=torch.FloatTensor((0.6,0.9))

print(loss_fn(a,b))
a=torch.FloatTensor([0.6,0.1])
b=torch.FloatTensor((0.6,0.9))

print(loss_fn(a,b))