Tensorflow2.1 updated code


#1

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