The code for Attentive Recurrent Comparators is here: https://github.com/pranv/ARC 24
It includes Omniglot data downloading and iterating scripts along with all the models proposed in the paper (the nets are written and trained with Theano).
I will try to submit a PR for torchvision.datasets.Omniglot if I find some time