I\'m trying to migrate this Tensorflow 1.1 code: 3.3 Omniglot Character set classification using Prototypical Network.ipynb, to Tensorflow 2.x.
Here it is my code: