Hi, I see that Facebook has a package for exactly that: and other useful stuff such as:
Why isn’t this an integral part of PyTorch?
Thanks,