how to use a single dataset to train multiple input model in tensorflow keras

got the answer

def map_func(x):
    return {
        'input_1': x.input_1,
        'input_2': tf.reshape(x.input_2,(-1,5,6))
    }, x.labels