9.1.3 简洁实现
这里的nn.GRU
只传入了参数num_inputs
和num_hiddens
,那么为什么没有num_outputs
呢?难道跟从零开始实现的代码一样,有num_inputs=num_outputs
吗?实际上不是的,我们的GRU只会输出隐状态,最终的输出状态是需要再连接一个全连接层什么的。我们去看d2l.RNNModel
的实现,会发现里面实现了这个全连接层的。综上nn.GRU
不用传入参数num_outputs
这里的nn.GRU
只传入了参数num_inputs
和num_hiddens
,那么为什么没有num_outputs
呢?难道跟从零开始实现的代码一样,有num_inputs=num_outputs
吗?实际上不是的,我们的GRU只会输出隐状态,最终的输出状态是需要再连接一个全连接层什么的。我们去看d2l.RNNModel
的实现,会发现里面实现了这个全连接层的。综上nn.GRU
不用传入参数num_outputs