1. Learn
    2. /
  2. Курси
    3. /
  3. Середній рівень Deep Learning з PyTorch
Connected

вправа

Мережа LSTM

Як ви вже знаєте, звичайні осередки RNN рідко використовують на практиці. Частіше застосовують Long Short-Term Memory, або LSTM, які набагато краще працюють із довгими послідовностями. У цій вправі ви самостійно збудуєте мережу LSTM!

Найважливіша відмінність у реалізації порівняно з RNN, яку ви створили раніше, полягає в тому, що LSTM має два, а не один прихований стани. Отже, вам потрібно ініціалізувати додатковий прихований стан і передати його до осередку LSTM.

torch і torch.nn уже імпортовано, тож почнімо кодувати!

Інструкції

100 XP
  • У методі .__init__() оголосіть шар LSTM і присвойте його до self.lstm.
  • У методі forward() ініціалізуйте перший прихований стан довготривалої пам'яті c0 нулями.
  • У методі forward() передайте до шару LSTM усі три вхідні значення: вхід для поточного кроку часу та кортеж із двох прихованих станів.