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]

flatten_parameters() [source]

Сбрасывает указатель на данные параметров, чтобы можно было использовать более быстрые пути кода.

В данный момент это работает только если модуль находится на видеокарте и включена cuDNN. В противном случае, это бесполезная операция.