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 (Tensor) – 输入

• dim (int) – 将计算 softmin 的维度（因此沿 dim 的每个切片将总和为 1）。

• dtype (torch.dtype, 可选) – 返回张量的期望数据类型。如果指定，则输入张量在执行操作之前被转换为 dtype。这对于防止数据类型溢出很有用。默认值：None。

返回类型

Tensor