在pytorch中,对于多分类问题我们使用 nn.CrossEntropyLoss()和nn.NLLLos等来计算softmax交叉熵
nn.CrossEntropyLoss()的输入要求是一个未激活的输入,会自动激活
torch 中 view()函数 修改tensor的形状
在pytorch中,对于多分类问题我们使用 nn.CrossEntropyLoss()和nn.NLLLos等来计算softmax交叉熵
nn.CrossEntropyLoss()的输入要求是一个未激活的输入,会自动激活
torch 中 view()函数 修改tensor的形状