Size must be a 1-D int32 Tensor

Question

I have MNIST data and doing some Transformation using Tensorflow and keras in R

 dim(train_images) <- c(nrow(train_images), 28,28,1)
 dim(test_images) <