rnn
# 第一个参数是输入维数,第二个参数是隐藏层维数,即输出维数,第三个参数是RNN的层数 rnn = nn.RNN(3, 10, 2) # 输入向量 第一个是序列长度,可以想象成句子的单词数,第二个是batch大小,第三个是每个词的维数,要和rnn第一个参数相同 x = torch.randn(1, 1, 3) # out是结果,h是隐藏层参数 out, h = rnn(x)
# 第一个参数是输入维数,第二个参数是隐藏层维数,即输出维数,第三个参数是RNN的层数 rnn = nn.RNN(3, 10, 2) # 输入向量 第一个是序列长度,可以想象成句子的单词数,第二个是batch大小,第三个是每个词的维数,要和rnn第一个参数相同 x = torch.randn(1, 1, 3) # out是结果,h是隐藏层参数 out, h = rnn(x)