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 (張量) – 輸入

• dim (int) – 將計算 softmax 的維度。

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

返回型別

張量

注意

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