Thanks, but I am still confused on how to modify the code 52 to keep pytorch Tensor.
I suppose I shall open a different thread for this gpu/cpu backend isue since the current thread topic seems to be have solved at least for now. I will confirm the solution later once I am done with the code modification.