Python PyTorch LogSoftmax用法及代码示例

本文简要介绍python语言中 torch.nn.LogSoftmax 的用法。

用法:

class torch.nn.LogSoftmax(dim=None)

参数：

dim(int) -将沿其计算LogSoftmax 的维度。

返回：

与输入具有相同维度和形状的张量，其值在 [-inf, 0) 范围内

将 \log(\text{Softmax}(x)) 函数应用于 n 维输入张量。 LogSoftmax 公式可以简化为：

\text{LogSoftmax}(x_{i}) = \log\left(\frac{\exp(x_i) }{ \sum_j \exp(x_j)} \right)

形状：

• 输入：(*) 其中 * 表示任意数量的附加维度

• 输出：(*)，与输入的形状相同

例子：

>>> m = nn.LogSoftmax()
>>> input = torch.randn(2, 3)
>>> output = m(input)