README.md

DeformableOffsets

产品支持情况

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

功能说明

  • 算子功能:用于计算变形卷积(Deformable Convolution)输出的函数。通过引入偏移参数offsets,使得卷积核在输入特征图上的位置可以动态调整,从而适配不规则的集合变化。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入的坐标。 DT_FLOAT16, DT_FLOAT, DT_BF16 NHWC
offsets 输入 偏移值。 DT_FLOAT16, DT_FLOAT, DT_BF16 NHWC
y 输出 变形卷积的输出。 DT_FLOAT16, DT_FLOAT, DT_BF16 NHWC
strides 属性 指定卷积核在高度和宽度方向上的步幅。 INT64 -
pads 属性 指定在输入的高度和宽度方向上添加的填充。 INT64 -
ksize 属性 指定卷积核的大小。 INT64 -
dilations 属性 指定每个维度上的膨胀因子。 INT64 -
data_format 属性 指定输入 x 的数据格式。 STRING -
deformable_groups 属性 指定输入 x 在 C 轴上的分组数。 INT32 -
modulated 属性 指定变形卷积的版本。 BOOL -

约束说明

调用说明

调用方式 样例代码 说明
图模式 test_geir_deformable_offsets 通过算子IR构图方式调用DeformableOffsets算子。