TensorFlow implementation of the Non-local Neural Network block in its various forms. - View it on GitHub