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, optional) – 返回張量所需的資料類型。 如果指定，則在執行操作之前，輸入張量會被轉換為 dtype。 這對於防止資料類型溢位非常有用。 預設值：None。

回傳類型

Tensor

注意

此函數無法直接與 NLLLoss 搭配使用，後者希望在 Softmax 和自身之間計算 Log。 請改用 log_softmax（它更快，並且具有更好的數值特性）。