https://locaakj.blogspot.com/2022/10/non-local-neural-networks-github.html