torch.nn.functional.log_softmax#
- torch.nn.functional.log_softmax(input, dim=None, _stacklevel=3, dtype=None)[源码]#
应用 softmax 后再应用对数。
虽然在数学上等同于 log(softmax(x))，但单独执行这两个操作会更慢且数值不稳定。此函数使用替代公式来正确计算输出和梯度。
有关更多详细信息，请参见
LogSoftmax。- 参数
input (张量) – 输入
dim (整型) – 将沿其计算 log_softmax 的维度。
dtype (
torch.dtype, 可选) – 返回张量所需的数据类型。如果指定，则在执行操作之前将输入张量转换为dtype。这有助于防止数据类型溢出。默认值：无。
- 返回类型