9.3.deep-rnn
符合直觉地,每一个隐状态依赖于 (前一时刻同一层 + 同一时刻前一层) 这两个隐状态。
高级 api 接口
仅需给出第三个参数 num_layers.
lstm_layer = nn.LSTM(num_inputs, num_hiddens, num_layers)
model = d2l.RNNModel(lstm_layer, len(vocab))
符合直觉地,每一个隐状态依赖于 (前一时刻同一层 + 同一时刻前一层) 这两个隐状态。
仅需给出第三个参数 num_layers.
lstm_layer = nn.LSTM(num_inputs, num_hiddens, num_layers)
model = d2l.RNNModel(lstm_layer, len(vocab))