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. В противном случае метод ничего не делает.