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)[source][source]¶
RNN 模組(RNN、LSTM、GRU)的基類。
實現了 RNN、LSTM 和 GRU 類共享的 RNN 方面的特性,例如模組初始化和用於引數儲存管理的實用方法。
注意
RNNBase 類沒有實現 forward 方法。
注意
LSTM 和 GRU 類覆蓋了 RNNBase 實現的一些方法。