the problem I face is that I am not sure what to do with THC/THC.h and the THCudaTensors it exposes. what are its PyTorch 1.0 counterparts?
example : https://github.com/peterhj/shiftnet_cuda_v2/blob/master/src/shiftnet_cuda.c 21