Pytorch，如何将CNN的输出输入到RNN的输入中？

import torch
import torch.nn as nn

class CNN(nn.Module):
    def __init__(self):
        super(CNN, self).__init__()
        # 定义CNN的结构

    def forward(self, x):
        # 实现CNN的前向传播
        return x

class RNN(nn.Module):
    def __init__(self):
        super(RNN, self).__init__()
        # 定义RNN的结构

    def forward(self, x):
        # 实现RNN的前向传播
        return x

cnn_model = CNN()
rnn_model = RNN()

cnn_output = cnn_model(input_data)

rnn_input = cnn_output.unsqueeze(0)  # 添加一个维度作为batch_size

rnn_output = rnn_model(rnn_input)