pytorch implementation

[smileyenot983]

Hi, colleagues, thanks for this work. A very interesting and unusual idea, I took your work as part of my project at University and reimplemented it in PyTorch.
Leaving here a link to my repo just in case someone feels more comfortable with PyTorch:
https://github.com/smileyenot983/shortcut_pytorch