TensorFlow implementation of 'A simple neural network module for relational reasoning' for bAbi task.