快捷方式

LogSoftmax

class torch.nn.LogSoftmax(dim=None)[source][source]

log(Softmax(x))\log(\text{Softmax}(x)) 函数应用于 n 维输入张量。

LogSoftmax 公式可以简化为

LogSoftmax(xi)=log(exp(xi)jexp(xj))\text{LogSoftmax}(x_{i}) = \log\left(\frac{\exp(x_i) }{ \sum_j \exp(x_j)} \right)
形状
  • 输入: ()(*) 其中 * 表示任意数量的附加维度

  • 输出: ()(*),与输入形状相同

参数

dim (int) – 计算 LogSoftmax 的维度。

返回

一个与输入维度和形状相同的张量,其值在 [-inf, 0) 范围内

返回类型

None

示例

>>> m = nn.LogSoftmax(dim=1)
>>> input = torch.randn(2, 3)
>>> output = m(input)

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

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

查看资源