torch.nn.LogSoftmax用法

LOGSOFTMAX

CLASS torch.nn.LogSoftmax(dim: Optional[int] = None)

\large log(Softmax(x))函数应用于n维输入张量。 LogSoftmax公式可以简化为:

\large \text{LogSoftmax}(x_{i}) = \log\left(\frac{\exp(x_i) }{ \sum_j \exp(x_j)} \right)

Shape:

  • Input: (∗) where ∗ means, any number of additional dimensions

  • Output: (∗) , same shape as the input

Parameters

dim (int) – A dimension along which LogSoftmax will be computed.用来计算LogSoftmax的维度。

Returns

a Tensor of the same dimension and shape as the input with values in the range [-inf, 0)。与输入具有相同维度和形状的张量,其值在[-inf,0)范围内。

Examples:

>>> m = nn.LogSoftmax()
>>> input = torch.randn(2, 3)
>>> output = m(input)
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值