Resising tensor in pytorch

vision
1

Pytorch transforms module has some functionality to resize images, which is great. However, I would like to resize an image which has already been transferred to the GPU.

Is there some way to perform the resize operation in pytorch on the GPU using linear interpolation?

2

I think you can use the interpolate function: https://pytorch.org/docs/stable/nn.html#torch.nn.functional.interpolate

1 Like
3

Ah thank you. Yes, I will give this a try.