Tensorflow implementation of image grid patches transform - View it on GitHub