def __init__(self, batch_size, input_size, args):
super(GRU_Decoder, self).__init__()
self.hidden_size = hidden_size = args.hidden_size
self.h_attention = Attention(hidden_size, args)
self.gru0 = nn.GRUCell(input_size=input_size, hidden_size=hidden_size)
self.gru1 = nn.GRUCell(input_size=hidden_size, hidden_size=hidden_size)
