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。

返回类型

张量