def __init__(self, ....):
  ....
  self.gru = nn.GRU(10, 204)
  self.dropout = nn.Dropout(0.5)
  self.fc = nn.Linear(204, 10)

def forward(self, x):
  x, hidden = self.gru(x)
  x = self.dropout(x)
  return self.fc(x)