Assume I have a multi-GPU system. Let tensor “a” be on one of the GPUs, and tensor “b” be on CPU. How can I move “b” to the same GPU that “a” resides?
Unfortunately, b.type_as(a) always moves b to GPU 0.
Thanks.