I get a dimension error which makes no sense “expected shape = ”

Question

So I have made the following implementation of the unet model:

def unet_model(optimizer, loss_metric, metrics, sample_width, sample_height, lr=1e-3):
    inputs =