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