RNNBase

class torch.nn.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 時有效。否則，它將不起作用。