Hi,
I am using topk on a cuda tensor and it is quite slow, so I suspect behind the scenes it copies the tensor to CPU and performs the topk over there then returning the result to GPU.
How can I affirm if this is the situation?
More generally, is there any indication in documentation as to which pytorch functionals will go forth and back GPU<>CPU so that I can avoid using them ?
Thanks,
Moshe