torch.nn.functional.softmin

torch.nn.functional.softmin(input, dim=None, _stacklevel=3, dtype=None)

應用 softmin 函式。

請注意，$\text{Softmin}(x) = \text{Softmax}(-x)$。數學公式請參見 softmax 定義。

更多詳情請參見 Softmin。

引數

• input (張量) – 輸入張量

• dim (int) – 計算 softmin 的維度（沿此維度的每個切片將求和為 1）。

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

返回型別

張量