pytorch实现CNN-LSTM

下面是一个简单的示例，展示了如何使用 PyTorch 实现一个基于 CNN-LSTM 的模型：

import torch
import torch.nn as nn

class CNN_LSTM(nn.Module):
    def __init__(self, input_dim, hidden_dim, output_dim, num_layers):
        super(CNN_LSTM, self).__init__()
        
        # 定义 CNN 层
        self.conv = nn.Conv2d(in_channels=1, out_channels=16, kernel_size=3, stride=1, padding=1)
        self.relu = nn.ReLU()
        self.pool = nn.MaxPool2d(kernel_size=2, stride=2)
        
        # 定义 LSTM 层
        self.lstm = nn.LSTM(input_dim, hidden_dim, num_layers, batch_first=True)
        
        # 定义全连接层
        self.fc = nn.Linear(hidden_dim, output_dim)
        
    def forward(self, x):
        # CNN 层的前向传播
        x = self.conv(x)
        x = self.relu(x)
        x = self.pool(x)
        
        # 将 CNN 输出的特征图转换为 LSTM 输入的序列
        x = x.view(x.size(0), -1, x.size(1) * x.size(2) * x.size(3))
        
        # LSTM 层的前向传播
        out, (h_n, c_n) = self.lstm(x)
        
        # 取 LSTM 最后一个时间步的输出作为全连接层的输入
        out = self.fc(h_n[-1])
        
        return out

在这个示例中，我们定义了一个名为 `CNN_LSTM` 的 PyTorch 模型，它由一个 CNN 层、一个 LSTM 层和一个全连接层组成。在前向传播中，我们首先对输入数据进行 CNN 处理，然后将 CNN 输出的特征图转换为 LSTM 输入的序列。接着，我们使用 LSTM 层对序列进行处理，并取 LSTM 最后一个时间步的输出作为全连接层的输入。最终，模型输出全连接层的输出。