def __init__(self, N, pool, utilities):
    batch_size = N.size()[0]
    self.N = N
    self.pool = pool
    self.utilities = torch.zeros(batch_size, 2, 3, dtype=torch.int64, device=FLAGS.device)
    self.utilities[:, 0] = utilities[0]
    self.utilities[:, 1] = utilities[1]
    self.last_proposal = torch.zeros(batch_size, 3, dtype=torch.int64, device=FLAGS.device)
    self.m_prev = torch.zeros(batch_size, FLAGS.utt_max_length, dtype=torch.int64, device=FLAGS.device)
def __init__(self, N, pool, utilities):
    batch_size = N.size()[0]
    self.N = N
    self.pool = pool
    self.utilities = torch.zeros(batch_size, 2, 3, dtype=torch.int64, device=FLAGS.device)
    self.utilities[:, 0] = utilities[0]
    self.utilities[:, 1] = utilities[1]
    self.last_proposal = torch.zeros(batch_size, 3, dtype=torch.int64, device=FLAGS.device)
    self.m_prev = torch.zeros(batch_size, FLAGS.utt_max_length, dtype=torch.int64, device=FLAGS.device)