torch.nn.functional.nll_loss
时间: 2023-04-25 10:00:05 浏览: 249
Pytorch中torch.nn的损失函数
5星 · 资源好评率100%
torch.nn.functional.nll_loss是PyTorch中的一个函数,用于计算负对数似然损失。它通常用于多分类问题中,其中每个样本只能属于一个类别。该函数的输入是一个张量,其中包含了每个样本的预测概率和真实标签。输出是一个标量,表示整个批次的平均损失。
阅读全文
Pytorch中torch.nn的损失函数