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 各個面向，例如模組初始化和用於參數儲存管理的實用方法。

注意

forward 方法未由 RNNBase 類別實作。

注意

LSTM 和 GRU 類別會覆寫 RNNBase 實作的一些方法。

flatten_parameters()

重設參數資料指標，以便它們可以使用更快的程式碼路徑。

目前，這僅在模組位於 GPU 上且啟用 cuDNN 時有效。 否則，它是一個空操作。