def get_inputs(batch_size=1):                                                                                                                                                                                                                 