
490 Глава 15
Форма b_xh: torch.Size([2])
Форма b_hh: torch.Size([2])
Вход этого слоя имеет форму
(batch_size, sequence_length, 5)
, где первое измерение —
это размер пакета (поскольку мы установили
batch_first=True
), второе измерение соот-
ветствует последовательности, а последнее измерение соответствует признакам. Для
входной последовательности длины 3 мы должны получить выходную последователь-
ность
012
,, .ooo Кроме того,
RNN
по умолчанию использует один слой, и вы можете
задать значение
num_layers
, чтобы получить несколько слоев и сформировать стековую
RNN.
Теперь выполним прямой проход для
rnn_layer
, вычислим выходные данные ...