pytorch多分类时，nn.CrossEntropyLoss()函数用法

在PyTorch中进行多分类任务时，可以使用`nn.CrossEntropyLoss()`函数来计算损失。`nn.CrossEntropyLoss()`函数结合了`nn.LogSoftmax()`和`nn.NLLLoss()`两个函数，可以同时完成softmax操作和交叉熵损失的计算。以下是`nn.CrossEntropyLoss()`函数的用法：

import torch.nn as nn

loss_fn = nn.CrossEntropyLoss()

# 假设模型输出为output，标签为target
loss = loss_fn(output, target)

其中，`output`是模型的输出，形状为`(batch_size, num_classes)`，表示每个样本属于每个类别的概率分布；`target`是标签，形状为`(batch_size,)`，表示每个样本的真实标签。

在调用`nn.CrossEntropyLoss()`函数时，它会自动对模型的输出进行softmax操作，并且将标签转换为整数形式。因此，我们不需要手动进行这些操作。损失值`loss`为一个标量，表示模型在这个batch上的平均交叉熵损失。

在训练过程中，我们可以通过反向传播求导来更新模型参数，使得损失值逐渐降低，最终使模型达到最优状态。