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。

Parameters

• 输入 (张量) – 输入

• dim (int) – 计算softmax的维度。

• dtype (torch.dtype, 可选) – 返回张量所需的数据类型。 如果指定，输入张量在操作执行前会被转换为dtype。这对于防止数据类型溢出很有用。默认值：None。

Return type

张量

注意

此函数不能直接与NLLLoss一起使用，因为NLLLoss期望在Softmax和自身之间计算Log。请改用log_softmax（它更快且具有更好的数值特性）。