Cannot change the dtype using tf.cast, when value passed is dataset.take(1)

Question

I want to change the dtype of one element of my dataset. (element shape = (32,28,28)--> this is one batch of 28 by 28 images in mnist dataset)

So i ran the followi