'Returns an inputs batch for the given event sequences.
Args:
event_sequences: A list of list-like event sequences.
full_length: If True, the inputs batch will be for the full length of
each event sequence. If False, the inputs batch will only be for the
last event of each event sequence. A full-length inputs batch is used
for the first step of extending the event sequences, since the RNN
cell state needs to be initialized with the priming sequence. For
subsequent generation steps, only a last-event inputs batch is used.
Returns:
An inputs batch. If `full_length` is True, the shape will be
[len(event_sequences), len(event_sequences[0]), INPUT_SIZE]. If
`full_length` is False, the shape will be
[len(event_sequences), 1, INPUT_SIZE].'

def get_inputs_batch(self, event_sequences, full_length=False):

    inputs_batch = []
    for events in event_sequences:
        inputs = []
        if full_length:
            for i in range(len(events)):
                inputs.append(self.events_to_input(events, i))
        else:
            inputs.append(self.events_to_input(events, (len(events) - 1)))
        inputs_batch.append(inputs)
    return inputs_batch
