LSTM和GRU原理及pytorch代码，输入输出大小说明