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。

引數

• input (Tensor) – 輸入

• dim (int) – Softmax 將沿其計算的維度。

• dtype (torch.dtype, 可選) – 返回張量的期望資料型別。如果指定，則在執行操作之前，輸入張量將轉換為 dtype。這對於防止資料型別溢位非常有用。預設值：None。

返回型別

Tensor

注意

此函式不能直接與 NLLLoss 一起使用，NLLLoss 需要在 Softmax 自身之間計算 Log。請改用 log_softmax（它更快且具有更好的數值特性）。