含有两层lstm的pytorch

含有两层LSTM的PyTorch模型可以通过`torch.nn.LSTM`类来实现。每一层LSTM都是相互独立的，其输入来自上一层的隐藏状态。在PyTorch中，可以通过设置`num_layers`参数来指定LSTM层数。以下是一个示例代码：

import torch
import torch.nn as nn

# 定义模型
class TwoLayerLSTM(nn.Module):
    def __init__(self, input_size, hidden_size, num_layers):
        super(TwoLayerLSTM, self).__init__()
        self.lstm = nn.LSTM(input_size, hidden_size, num_layers)

    def forward(self, x):
        # 前向传播
        output, _ = self.lstm(x)
        return output

# 创建模型实例
input_size = 100
hidden_size = 200
num_layers = 2
model = TwoLayerLSTM(input_size, hidden_size, num_layers)

# 打印模型结构
print(model)

# 输出:
# TwoLayerLSTM(
#   (lstm): LSTM(100, 200, num_layers=2)
# )