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 的维度（因此沿 dim 的每个切片将总和为 1）。

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

返回类型

张量