Imported from https://github.com/schatty/matching-networks-tf