1. 函数介绍
创建一个衡量目标和输出之间二进制交叉熵的标准。
注意:一般使用之前需要使用激活函数将数据处理成0到1之间,一般使用sigmoid函数。
2. 原理理解
3. 代码
import torch
x = torch.randn(size=[2, 2])
sig = torch.nn.Sigmoid()
out_ = sig(x)
target = torch.FloatTensor([[1, 0], [0, 1]])
loss = torch.nn.BCELoss()
loss_ = loss(out_, target)
print(loss_)