代码:torch.nn.CrossEntropyLoss()
定义:loss_cross = torch.nn.CrossEntropyLoss()
使用:loss = loss_cross(out, y)
参数解释:
**out:**网络输出的结果,且该结果经过了sigmoid
,形状为【N,class_num】
y::
代码:torch.nn.CrossEntropyLoss()
定义:loss_cross = torch.nn.CrossEntropyLoss()
使用:loss = loss_cross(out, y)
参数解释:
**out:**网络输出的结果,且该结果经过了sigmoid
,形状为【N,class_num】
y::