torch.nn.functional.softmin¶
- torch.nn.functional.softmin(input, dim=None, _stacklevel=3, dtype=None)[source]¶
应用 softmin 函数。
请注意,. 请参阅 softmax 定义以了解数学公式。
有关详细信息,请参阅
Softmin
。- 参数
input (Tensor) – 输入
dim (int) – 将计算 softmin 的维度(因此沿 dim 的每个切片之和为 1)。
dtype (
torch.dtype
, 可选) – 返回张量的所需数据类型。如果指定,则在执行操作之前将输入张量转换为dtype
。这对于防止数据类型溢出很有用。默认值:无。
- 返回类型