I am thinking to create the Patch Match algorithm using Pytorch. The idea behind the implementation is to use Tensors with GPU support in order to do the calculations fast. A fast overview of patch match algorithm is the following:
Patch match is a randomized algorithm that tries to approximate nearest neighbors of a reference patch in a target set of patches. Here is a video explanation: https://www.youtube.com/watch?v=fMe19oTz6vk (4 min)
Do you think that a Pytorch implementation will be feasible? Also, this algorithm make heavy use of indices so I am not sure if the end result will be fast.