Grid in pytorch_ How to use sample

https://blog.csdn.net/qq_34914551/article/details/107559031

import torch
from torch.nn import functional as F


inp = torch.ones(1, 1, 4, 4)

#  The goal is to get a   Length and width are 20 Of tensor
out_h = 20
out_w = 20
 # grid The generation method of is equivalent to mesh_grid
 #  This is generated directly [-1,1] Of grid coordinate , In practical application, it is necessary to unify the corresponding pixel coordinates of the image to [-1,1]
new_h = torch.linspace(-1, 1, out_h).view(-1, 1).repeat(1, out_w)
print("new_h shape:",new_h.shape)
print(new_h) # new_h shape: torch.Size([20, 20])
new_w = torch.linspace(-1, 1, out_w).repeat(out_h, 1)
print("new_w shape:", new_w.shape)
print(new_w) #new_w shape: torch.Size([20, 20])
grid = torch.cat((new_h.unsqueeze(2), new_w.unsqueeze(2)), dim=2)
print(grid.shape) #torch.Size([20, 20, 2])
grid = grid.unsqueeze(0)
print(grid.shape) # torch.Size([1, 20, 20, 2])
outp = F.grid_sample(inp, grid=grid, mode='bilinear')
print(outp.shape)  #torch.Size([1, 1, 20, 20])