GridUnnormal

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	√

功能说明

• 算子功能：GridUnnormal是GridSample链路中的坐标反归一化算子。输入归一化坐标grid和同shape的尺寸辅助张量assist，输出小数偏移diff和整数采样位置position。

• 计算公式：对每个元素独立计算：

  t        = (grid + 1) * 0.5
  pos_base = align_corners ? t * (assist - 1) : t * assist - 0.5
  position = floor(pos_base)
  diff     = pos_base - floor(pos_base)
  

floor按向负无穷取整语义执行，不能用向零截断替代。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
grid	输入	归一化采样坐标。支持4D静态shape、动态shape和编译期未知rank；rank确定时必须为4且末维为2；支持总元素数为0的空Tensor。	float16、float32	ND
assist	输入	每元素对应的输入尺寸辅助值。必须与gridshape、dtype完全一致，不支持广播；rank确定时必须为4且末维为2。	float16、float32	ND
diff	输出	pos_base的小数部分，shape与grid一致，dtype与grid一致。	float16、float32	ND
position	输出	pos_base的下取整结果，shape与grid一致。	int32	ND
align_corners	属性	可选属性，默认false。为true时按t * (assist - 1)计算；为false时按t * assist - 0.5计算。	bool	-

约束说明

• grid与assist的shape、dtype必须一致。
• rank确定时，grid与assist必须为4D Tensor，shape为[batch, height, width, 2]。
• 仅支持ND格式；非连续Tensor作为用户可见接口不涉及。
• diff的dtype跟随grid；position固定为int32。
• 中间计算使用fp32；fp16输入会提升到fp32计算，diff再回写为fp16。
• 总元素数为0的空Tensor支持空进空出，设备侧不访问数据。
• 非有限输入以及floor(pos_base)超出int32表示范围不属于本算子支持域。
• 本算子为GE图内部算子，不提供aclnn、torch、TensorFlow、ONNX、Caffe对外接口。

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_grid_unnormal	通过本目录的算子原型构图方式调用GridUnnormal算子。