RNNBase

class torch.nn.modules.rnn.RNNBase(mode, input_size, hidden_size, num_layers=1, bias=True, batch_first=False, dropout=0.0, bidirectional=False, proj_size=0, device=None, dtype=None)

RNN 模块（RNN、LSTM、GRU）的基类。

实现了 RNN、LSTM 和 GRU 类共有的 RNN 的一些方面，例如模块初始化和用于参数存储管理的实用方法。

注意

RNNBase 类没有实现 forward 方法。

注意

LSTM 和 GRU 类会覆盖 RNNBase 实现的一些方法。

flatten_parameters()

重置参数数据指针，以便它们可以使用更快的代码路径。

目前，这仅在模块位于 GPU 上且启用了 cuDNN 时才有效。否则，它将不起作用。