My function use these data(generated from the GPUs) for Backpropagation training. I use a third lib(Ray) to generate the cuda data and when I return them to the main function, there is an error “TypeError: can’t convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.” So I think I can’t return the cuda tensor directly.