LogSoftmax

类 torch.nn.LogSoftmax(dim=None)[source][source] ¶

将 $\log(\text{Softmax}(x))$ 函数应用于 n 维输入张量。

The LogSoftmax 公式可以简化为：

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

形状：

• 输入： $(*)$ 其中 * 表示任意数量的额外维度

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

参数:

dim（int）- LogSoftmax 将计算的维度。

返回值:

与输入具有相同维度和大小的 Tensor，其值在范围[-inf, 0)内。

返回类型:

无。

示例：

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