def __init__(self, embed_size=512):
    super(GRU, self).__init__()
    self.update_x = nn.Linear(embed_size, embed_size, bias=True)
    self.update_h = nn.Linear(embed_size, embed_size, bias=True)
    self.reset_x = nn.Linear(embed_size, embed_size, bias=True)
    self.reset_h = nn.Linear(embed_size, embed_size, bias=True)
    self.memory_x = nn.Linear(embed_size, embed_size, bias=True)
    self.memory_h = nn.Linear(embed_size, embed_size, bias=True)
def __init__(self, embed_size=512):
    super(GRU, self).__init__()
    self.update_x = nn.Linear(embed_size, embed_size, bias=True)
    self.update_h = nn.Linear(embed_size, embed_size, bias=True)
    self.reset_x = nn.Linear(embed_size, embed_size, bias=True)
    self.reset_h = nn.Linear(embed_size, embed_size, bias=True)
    self.memory_x = nn.Linear(embed_size, embed_size, bias=True)
    self.memory_h = nn.Linear(embed_size, embed_size, bias=True)