When I create an InputLayer, a None dimension gets added. I don't know why this is.
image_shape = ds_info.features['image'].shape
input_shape = (ds_info.features['label'].num_classes, image_shape[0], image_shape[1], image_shape[2])
print('input_shape: ', input_shape)
## Create the CNN
# Create the input layer
input_layer = Input(shape=input_shape)
print('input_layer: ', input_layer.shape)
Results
input_shape: (1623, 105, 105, 3)
input_layer: (None, 1623, 105, 105, 3)