代码如下: #自定义LSTM实现 class NaiveCustomLSTM(nn.Module): def __init__(self,input_size,hidden_size,num_layers=2):