Pythorch中torch.nn.LSTM()参数详解

通过源代码中可以看到nn.LSTM继承自nn.RNNBase,其初始化函数定义如下 class RNNBase(Module): ... def __init__(self, mode, input_size, hidden_size, num_layers=1, bias=True, batch_first=False, dropout=0., bidirectional=False): 我们需要关注的参数以及其含义解释如下: input_size – 输入数据的大小,也就是前面例子中每个单词向量的