torch.nn.functional.log_softmax¶
- torch.nn.functional.log_softmax(input, dim=None, _stacklevel=3, dtype=None)[原始碼][原始碼]¶
應用 softmax 後接一個對數。
雖然數學上等同於 log(softmax(x))，但分別執行這兩個操作會更慢且數值不穩定。此函式使用替代公式來正確計算輸出和梯度。
有關更多詳細資訊，請參閱
LogSoftmax。- 引數
input (Tensor) – 輸入張量
dim (int) – 計算 log_softmax 的維度。
dtype (
torch.dtype, 可選) – 返回張量的期望資料型別。如果指定，將在執行操作前將輸入張量轉換為dtype。這對於防止資料型別溢位非常有用。預設值：None。
- 返回型別
