文件最后提交记录最后更新时间
7 个月前
2 个月前
2 个月前
1 个月前
2 个月前
1 个月前
3 个月前
4 个月前
README

MaxPoolGradWithArgmaxV3

产品支持情况

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

功能说明

  • 算子功能:正向最大池化MaxPoolGradWithArgmaxV3的反向。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入x FLOAT16、BFLOAT16、FLOAT ND
grad 输入 梯度Tensor FLOAT16、BFLOAT16、FLOAT ND
argmax 输入 输入grad INT32 ND
ksize 输入属性 池化窗口大小 Int ND
strides 输入属性 窗口移动步长 Int ND
pads 输入属性 每一条边补充的层数 Int ND
dtype 输入属性 输出argmax的数据类型 Int ND
dilation 输入属性 控制窗口中元素步幅 Int ND
ceil_mode 输入属性 为true是用向上取整的方法 Bool ND
data_format 输入属性 数据格式 String ND
y 输出 输出 FLOAT16、BFLOAT16、FLOAT ND

约束说明

调用说明

调用方式 样例代码 说明
aclnn调用 test_aclnn_max_pool_grad_with_argmax_v3 通过aclnnMaxPool2dWithIndicesBackward接口方式调用aclnnMaxPoolGradWithArgmaxV3算子。