本文共 381 字，大约阅读时间需要 1 分钟。

import torchimport torch.nn as nn# rnn第一个参数是输入特征维度，第二个是隐藏层特征维度，第三个是隐藏层层数rnn = nn.RNN(5, 6, 2)# 输入，第一个表示sequence_length，第二个表示batch_size，第三个表示特征维度，即tensor的维度input = torch.randn(1, 4, 5)# 第一个表示隐藏层的数量，第二个表示batch_size，第三个表示隐藏层特征维度h0 = torch.randn(2, 4, 6)output, hn = rnn(input, h0)print(output)print(hn)

之前认知错误，认为输入第一个维度为batchsize， 第二个维度是sequence_length

转载地址：http://pvnws.baihongyu.com/