Pytorch implementaion

[umgpy]

Hi I was looking for a PyTorch implementation of the project. I'm willing to translate the code myself if there is no implementation.
Just wanted to know how can I go about it.
Cheers

[jiaoyining]

The main part of the algorithm is implemented with CUDA. Only cuda_op_kernel.cu.cc, which is for connecting the cuda part and the python part, is using tensorflow. Pytorch can read the .so file but not I haven't tried whether pytorch has corresponding functions to replace the tf functions in cuda_op_kernel.cu.cc.