torch.nn.functional.log_softmax¶
- torch.nn.functional.log_softmax(input, dim=None, _stacklevel=3, dtype=None)[source][source]¶
应用 softmax 后跟对数运算。
虽然数学上等同于 log(softmax(x))，但分别执行这两个操作会更慢且数值上不稳定。此函数使用一种替代公式来正确计算输出和梯度。
更多详情请见
LogSoftmax。- 参数:
输入（张量）- 输入
dim（整型）- log_softmax 将要计算的维度。
dtype（
torch.dtype，可选）- 返回张量的期望数据类型。如果指定，则在执行操作之前将输入张量转换为dtype。这有助于防止数据类型溢出。默认：None。
- 返回类型:
