Мережа GRU

Поряд із LSTM, ще одним популярним різновидом рекурентних нейронних мереж є Gated Recurrent Unit, або GRU. Її привабливість — у простоті: комірки GRU потребують менше обчислень, ніж комірки LSTM, водночас часто не поступаючись їм за якістю.

Наданий вам код — це визначення моделі RNN, яке ви писали раніше. Ваше завдання — адаптувати його так, щоб замість цього створити мережу GRU. torch і torch.nn as nn вже імпортовано.