torch.nn.functional.log_softmax¶
- torch.nn.functional.log_softmax(input, dim=None, _stacklevel=3, dtype=None)[源代码][源代码]¶
应用 softmax 函数，然后应用对数函数。
虽然数学上等价于 log(softmax(x))，但分别执行这两个操作速度较慢且数值不稳定。此函数使用另一种公式来正确计算输出和梯度。
有关更多详细信息，请参阅
LogSoftmax。- 参数
input (Tensor) – 输入
dim (int) – 将计算 log_softmax 的维度。
dtype (
torch.dtype, 可选) – 返回张量的期望数据类型。如果指定，则在执行操作之前，输入张量将转换为dtype。这对于防止数据类型溢出很有用。默认值：None。
- 返回类型
