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

MaxPoolGradWithArgmax

产品支持情况

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

功能说明

  • 算子功能:正向最大池化MaxPoolWithArgmax的反向梯度计算。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入x,形状为[NCHW]或[NHWC] FLOAT16、BFLOAT16、FLOAT NCHW、NHWC
grad 输入 梯度Tensor,形状为[NCHW]或[NHWC] FLOAT16、BFLOAT16、FLOAT NCHW、NHWC
argmax 输入 正向最大池化输出的索引,形状与grad相同 INT32、INT64 NCHW、NHWC
ksize 属性(必选) 池化窗口大小,长度为4的列表。NCHW格式时ksize[0]=1且ksize[1]=1;NHWC格式时ksize[0]=1且ksize[3]=1 ListInt ND
strides 属性(必选) 窗口移动步长,长度为4的列表。NCHW格式时strides[0]=1且strides[1]=1;NHWC格式时strides[0]=1且strides[3]=1 ListInt ND
padding 属性(必选) 填充算法,取值为"SAME"或"VALID" String ND
include_batch_in_index 属性(可选) 是否在计算argmax索引时包含batch维度。当前仅支持false。默认值:false Bool ND
data_format 属性(可选) 数据布局格式,取值为"NHWC"或"NCHW"。默认值:"NHWC" String ND
y 输出 输出梯度,形状与x相同 FLOAT16、BFLOAT16、FLOAT NCHW、NHWC

约束说明

  • include_batch_in_index:当前仅支持false

调用说明

调用方式 样例代码 说明
图模式接口 test_geir_max_pool_grad_with_argmax 通过IR MaxPoolGradWithArgmax构图方式调用MaxPoolGradWithArgmax算子。