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