I am using: torch.nn.functional.grid_sample
on the GPU however, I would like to use this operation on the CPU as well. Is there any way to do this?
You should be able to use the same function of you pass CPU-Tensors.
I am using: torch.nn.functional.grid_sample
on the GPU however, I would like to use this operation on the CPU as well. Is there any way to do this?
You should be able to use the same function of you pass CPU-Tensors.