循环神经网络:RNN层的定义

import torch

batch_size=1
seq_len=3
input_size=4  #独立x的列数
hidden_size=2
num_layers=1

cell=torch.nn.RNN(input_size=input_size,hidden_size=hidden_size,num_layers=num_layers)

inputs=torch.randn(seq_len,batch_size,input_size)
hidden=torch.zeros(num_layers,batch_size,hidden_size)

out,hidden=cell(inputs,hidden)

print(out)#中间运算出的h1~h(n-1)
print(hidden)#最后一个h(n)