python中lstm模型代码

以下是一个使用PyTorch实现LSTM模型的示例代码：

import torch
import torch.nn as nn

class LSTM(nn.Module):
    def __init__(self, input_size, hidden_size, output_size):
        super(LSTM, self).__init__()
        self.hidden_size = hidden_size
        self.lstm = nn.LSTM(input_size, hidden_size)
        self.fc = nn.Linear(hidden_size, output_size)

    def forward(self, input):
        h0 = torch.zeros(1, input.size(1), self.hidden_size)
        c0 = torch.zeros(1, input.size(1), self.hidden_size)
        output, (hn, cn) = self.lstm(input, (h0, c0))
        output = self.fc(output[-1])
        return output

这个LSTM模型包含一个LSTM层和一个全连接层。在初始化函数中，我们定义了输入大小，隐藏大小和输出大小。在前向函数中，我们首先将输入传递给LSTM层，然后将LSTM层的输出传递给全连接层。最后，我们返回全连接层的输出。