快捷键

torch.nn.functional.softmin

torch.nn.functional.softmin(input, dim=None, _stacklevel=3, dtype=None)[source][source]

应用软最大函数。

注意查看 Softmin(x)=Softmax(x)\text{Softmin}(x) = \text{Softmax}(-x) 。请参阅 softmax 定义中的数学公式。

更多详情请见 Softmin

参数:
  • 输入(张量)- 输入

  • dim(int)- 沿着该维度将计算 softmax(因此沿 dim 的每个切片都将求和为 1)。

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

返回类型:

张量


© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建,并使用 Read the Docs 提供的主题。

文档

PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源