获取_rnn#

pytorch_forecasting.models.nn.rnn.get_rnn(cell_type: Type[递归神经网络] | str) Type[递归神经网络][来源]#

获取 LSTM 或 GRU。

Parameters:

cell_type (联合[RNN, 字符串]) – “LSTM” 或 “GRU”

Returns:

返回 GRU 或 LSTM RNN 模块

Return type:

类型[RNN]