Imported from https://github.com/mari-linhares/tensorflow-maml