torch.nn.functional.log_softmax

torch.nn.functional.log_softmax(input, dim=None, _stacklevel=3, dtype=None)[源代码]

先应用softmax函数，然后取对数。

虽然在数学上等同于 log(softmax(x))，但是分别执行这两个操作会更加缓慢且数值不稳。此函数采用另一种方法来正确计算输出和梯度。

更多详情请参见LogSoftmax。

参数

• 输入 (Tensor) – 输入张量

• dim (int) – 指定用于计算对数软最大值的维度。

• dtype (torch.dtype, 可选) – 返回张量所需的数据类型。如果指定了 dtype，则在执行操作之前将输入张量转换为指定的 dtype，以防止数据溢出。默认值：None。

返回类型

Tensor