快捷方式

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 實現的一些方法。

flatten_parameters()[source][source]

重置引數資料指標,以便它們可以使用更快的程式碼路徑。

目前,這僅當模組在 GPU 上且啟用了 cuDNN 時才有效。否則,它是一個空操作。

文件

獲取 PyTorch 全面的開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲取問題解答

檢視資源