def init_hidden_state(self, batch, s_embed=256):
    init_s = torch.zeros(batch, s_embed).cuda()
    return init_s
def init_hidden_state(self, batch, s_embed=256):
    init_s = torch.zeros(batch, s_embed).cuda()
    return init_s