grid_sampler

    该OP基于flow field网格的对输入X进行双线性插值采样。网格通常由affine_grid生成, shape为[N, H, W, 2],是shape为[N, H, W]的采样点张量的(x, y)坐标。 其中,x坐标是对输入数据X的第四个维度(宽度维度)的索引,y坐标是第三维度(高维度)的索引,最终输出采样值为采样点的4个最接近的角点的双线性插值结果,输出张量的shape为[N, C, H, W]。

    step 1:

    step 2:

    参数:

    返回: Variable(Tensor): 输入X基于输入网格的双线性插值计算结果,维度为 [N,C,H,W][N,C,H,W] 的4-D Tensor

    代码示例: