RoiAlignRotatedGrad

产品支持情况

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

功能说明

• 算子功能：通过旋转框各点坐标将梯度回传至对应位置。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x_grad	输入	输入特征图的梯度。	FLOAT32	ND
rois	输入	ROI边界框。	FLOAT32	ND
spatial_scale	属性	ROI边界框的缩放率。	FLOAT32	-
sampling_ratio	属性	计算RoiAlign时的采样率。	INT	-
y_grad_shape	属性	输出的shape。	LISTINT	-
pooled_h	属性	RoiAlign输出时池化特征图的高。	INT	-
pooled_w	属性	RoiAlign输出时池化特征图的宽。	INT	-
aligned	属性	是否量化，true则将目标框中心坐标值减0.5。	BOOL	-
clockwise	属性	时钟方向，true为顺时针。	BOOL	-
y_grad	输出	输出特征图。	FLOAT32	ND

约束说明

• 输入x_grad的shape必须为[N,C,pooled_h,pooled_w]的Tensor。C的取值在[0, 1024]之间，pooled_h、pooled_w的取值在[0, 1024]之间。
• rois的shape必须为[n, 6]的Tensor。

调用说明

调用方式	样例代码	说明
图模式	test_geir_roi_align_rotated_grad	通过图模式方式调用RoiAlignRotatedGrad算子。