torch.nn.LogSoftmax

torch.nn.LogSoftmax是PyTorch中的一个类，用于计算输入张量的对数softmax函数。对数softmax函数是softmax函数的一种变体，它将输入张量的每个元素转换为其对数概率值。

对数softmax函数的计算公式如下：
logsoftmax(x_i) = log(exp(x_i) / sum(exp(x_j)))

其中，x_i表示输入张量的第i个元素，exp表示指数函数，sum表示求和函数。

torch.nn.LogSoftmax类的主要作用是将输入张量的每个元素进行对数softmax变换，并返回结果。它可以应用于多维张量，保持输入张量的形状不变。

使用torch.nn.LogSoftmax类可以通过以下方式实现：

import torch
import torch.nn as nn

# 创建LogSoftmax对象
log_softmax = nn.LogSoftmax(dim=1)

# 输入张量
input_tensor = torch.randn(2, 3)

# 对输入张量进行对数softmax变换
output_tensor = log_softmax(input_tensor)

print(output_tensor)