Tensorflow2.1 updated code

The code provided is in older version of tensor flow the updated one is 2.1 .Can you please help me with the ANN example of mnist to implement in 2.1