fix issues with deprecated library THC

[rram12]

First of all, thank you @tianzhi0549 for this great framework, it became quickly my favorite framework. I found that the THC library is deprecated for PyTorch > 1.6.0, and many of its methods are replaced by methods in the ATen library.