pytorch 中的 CrossEntropyLoss 与 NLLLoss
官网是这么介绍CrossEntropyLoss 的:
This criterion combines LogSoftmax and NLLLoss in one single class.
它将 LogSoftmax 与 NLLLoss 结合在单个类中.
用相应的函数进行验证:
import torch
import torch.nn.functional as F
N = 32
num_class = 1000
input = torch.randn(N, num_class)
targe.
原创
2021-07-27 14:33:23 ·
372 阅读 ·
0 评论