Bad tensor shape when using tensorflow drop_remainder

Question

I am trying to create a tf.data.Dataset from a generator. I want to make sure all of my batches have the exact same size, so I\'m calling .batch(batch_siz