how to write mnist.train.next_batch(50) in tensoeflow 2?

Question

how to use dataset.train.next_batch() in my tensor-flow 2 in my own data-set? i have searched a lot,But in-vain.

code code is:

  batch