GridSample

功能

该操作通过基于流场网格的插值对输入x进行采样，通常由affine_grid生成。形状为【N,H,W,2】的网格是（x,y）与形状【N,H,W】各坐标，其中x是索引的第4个维度（在宽度维度中）输入数据x和y是索引到第3维（在高度维度上），最终结果为4个最近角点的插值值。输出tensor形状将是【N,C,H,W】。

输入

两个输入：

x：四维Tensor，形状为[batch, channels, height, width]，数据类型：float16、float32。

grid：四维Tensor，形状为[batch, height, width, 2]，数据类型：float16、float32。

输出

一个输出：

y：四维Tensor，数据类型：float16, float32。

属性

interpolation_mode：指定插值方法的可选字符串。目前只支持bilinear和nearest，数据类型：string。

padding_mode：指定填充方法的可选字符串。数据类型：string。

align_corners：一个可选的bool。如果为“true”，则输入和输出tensor的角点像素的中心对齐。默认为“false”，数据类型：bool。

支持的ONNX版本

Opset v16