torch.nn.functional.softmin
- torch.nn.functional.softmin(input, dim=None, _stacklevel=3, dtype=None)[源代码]
-
应用软最小值函数。
注意 $\text{Softmin}(x) = \text{Softmax}(-x)$。查看softmax的定义以获取数学公式。
更多详情请参见
Softmin。- 参数
-
-
输入 (Tensor) – 输入张量
-
dim (int) – 沿此维度计算软最小值（因此沿着该维度的每个切片的和都将为1）。
-
dtype (
torch.dtype, 可选) – 返回张量的数据类型。如果指定了dtype，则在执行操作之前将输入张量转换为指定的dtype，以防止数据溢出。默认值：None。
-
- 返回类型