tf.data.Dataset iterator returning Tensor(“IteratorGetNext:1”, shape=(None, 16), dtype=int32) but cannot get the values of the Tensors

Question

I am trying to write a Custom Model in which I am writing a custom train_step function

I am creating a \'tf.data.Dataset` from a Custom Datagenerator like