Imported from https://github.com/ash3n/Prototypical-Network-TF