Amrit_Das:

TypeError: can’t convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.


In this current context, are there option of not copying the tensor data to host (cpu) memory ?
Please advise.