Pytorch warp on GPU

vision
1

Is there some equivalent to skimage.transform.warp function implemented for gpu?

2

Combine
torch.nn.functional. affine_grid ( theta , size )
and
torch.nn.functional. grid_sample ( input , grid , mode=‘bilinear’ , padding_mode=‘zeros’ )
you can achieve the equivalent function to skimage.transform.warp function.