"""
本文件包含该任务的模型
"""

from torch import nn
import torch

from config import *


class MyRnn(nn.Module):
    """ 基于RNN来预测 """
    def __init__(self, hidden_size: int) -> None:
        super().__init__()
        pass

    def forward(self, x):
        pass
        return x


if __name__ == '__main__':
    # 测试
    model = MyRnn(100)
    x = torch.rand((5, 3, 52))
    y = model(x)
    print(y.shape)