You won’t be able to directly pass CUDATensors from PyTorch to this method, since sklearn uses numpy as the backend, which cannot use the GPU.
However, you could check the implementation and rewrite this method using torch functions, which could then use CUDATensors.