remove output_shape from layer and infer the dimensions starting from the Inputs (investigate if possible)
remove output_shape from layer and infer the dimensions starting from the Inputs (investigate if possible)