Pytorch implementation of "A simple neural network module for relational reasoning" (Relational Networks) - View it on GitHub