【Pytorch实用教程】nn.LogSoftmax的详细用法及公式

nn.LogSoftmax 是 PyTorch 中的一个层，用于在对数空间中计算 Softmax 激活函数。它通常用于多类别分类任务中，以提高数值稳定性。其输入是 logits，即未归一化的得分，输出是这些得分在对数空间中的 Softmax。

用法

nn.LogSoftmax 可以在构建模型时作为层使用，也可以通过函数调用使用。

作为层使用

import torch
import torch.nn as nn

log_softmax = nn.LogSoftmax(dim=1)
input = torch.tensor([