torch.nn.functional.softmax
- torch.nn.functional.softmax(input, dim=None, _stacklevel=3, dtype=None)[源代码]
-
使用softmax函数。
softmax定义如下：
$\text{Softmax}(x_{i}) = \frac{\exp(x_i)}{\sum_j \exp(x_j)}$
它会应用到dim维度上所有的切片，并将其重新缩放，使元素的值在[0, 1]范围内且总和为1。
更多详情请参见
Softmax。- 参数
-
-
输入 (Tensor) – 输入张量
-
dim (int) – 计算 softmax 的维度。
-
dtype (
torch.dtype, 可选) – 返回张量的数据类型。如果指定了dtype，则在执行操作之前将输入张量转换为指定的dtype，以防止数据溢出。默认值：None。
-
- 返回类型
注意
此函数无法直接与NLLLoss一起使用，因为NLLLoss要求在Softmax和其本身之间进行Log计算。建议使用log_softmax（它更快且具有更好的数值稳定性）。