weixin_43896235的博客

CrossEntropy 输入：（样本数，Class 数）CrossEntropy 会计算所有样本loss，返回平均lossoutputs = torch.tensor([[0.9,0.1]]) #两个Class预测概率为0.9,0。1， 一个样本为[0.9,0.1]targets = torch.tensor([0]) #样本target为0，也就是第一个0.9loss(outputs,targets) #返回loss，如果多个样本返回平均loss ...