Your input images should have the mentioned shape ([batch_size, channels, height, width]), while they are 2-dimensional tensors.
Have a look at this tutorial 4 for an example on how to build a model and how the data tensors should be created.