基于pytorch，求解建立lstm分类模型思路和方法

有一个19200*303的数据集，要求利用pytorch建立一个lstm的五分类模型。
预测Loc HRR Wind的值（五分类）
初学，还理不太清模型的结构和相应的维度。

batch=30
步长=15
希望帮忙理一理思路，或者推荐相似的案例学习（当然有代码最好不过了😁

1、准备数据集
2、建立LSTM模型
3、loss和优化
4、训练模型

主要是不太清楚 LSTM模型中具体要用哪些结构和相应的维度