ExtendConvTranspose

产品支持情况

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

功能说明

• 算子功能：计算三维卷积的转置（反卷积）相对于输入的梯度。

• 计算公式：

  假定输入的shape为($N,C_{in},D_{in},H_{in},W_{in}$)、输出的shape为($N,C_{out},D_{out},H_{out},W_{out}$)，那么他们与卷积步长($stride$)、卷积核大小($kernel\_size，kD,kH,kW$)、膨胀参数($dilation$)的关系是：

  $$D_{out}=(D_{in}-1)\ast stride[0]-2\ast padding[0]+dilation[0]\ast (kernel\_size[0]-1)+output\_padding[0]+1$$

  $$H_{out}=(H_{in}-1)\ast stride[1]-2\ast padding[1]+dilation[1]\ast (kernel\_size[1]-1)+output\_padding[1]+1$$

  $$W_{out}=(W_{in}-1)\ast stride[2]-2\ast padding[2]+dilation[2]\ast (kernel\_size[2]-1)+output\_padding[2]+1$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
input_size	输入	1D张量，用于表示输入张量的形状。	INT32、INT64	-
x	输入	特征图，相当于公式中的($N,C_{in},D_{in},H_{in},W_{in}$)。 一个张量，数据格式与'y'一致。	FLOAT16、FLOAT32、BFLOAT16、INT8	NCDHW
filter	输入	一个5D张量，代表卷积核尺寸，相当于公式中的kernelSize[0]、kernelSize[1]、kernelSize[2]。	FLOAT16、FLOAT32、BFLOAT16、INT8	NCDHW、NDHWC
bias	可选属性	一个1D张量，卷积偏置张量。	FLOAT16、FLOAT32、INT32	ND
scale	可选属性	一个1D张量，用于量化反卷积，当前未启用。	UINT64	ND
strides	必填属性	一个包含5个整数的元组或列表，用于指定滑动窗口在输入张量'x'每个维度上的步长。 轴顺序与特征图格式一致。	-	-
pads	必填属性	一个包含6个整数的元组或列表，用于指定各方向的填充量，相当于公式中的padding[0]、padding[1]、padding[2]。	-	-
dilations	可选属性	一个包含5个整数的元组或列表，表示输入各维度的膨胀（空洞）因子，相当于公式中的dilation[0]、dilation[1]、dilation[2] 默认值为[1,1,1,1,1]。 轴顺序与特征图格式一致。	-	-
groups	可选属性	整数，范围为[1,65535]，默认1。 表示从$c_{in}$到$c_{out}$的分组连接数。 $c_{in}$与$c_{out}$必须能被'groups'整除。 不同'groups'与dtype的组合支持见下方表格说明。	INT	-
data_format	可选属性	字符串，取值必须为["NDHWC","NCDHW"]之一，默认"NDHWC"。对应关系为：batch(N)、depth(D)、height(H)、width(W)、channels(C)。 指定'x'与'y'的数据排布格式。	STRING	-
output_padding	可选属性	将在输出形状末尾额外增加的尺寸，默认值为[0,0,0,0,0]。 相当于公式中的output_padding[0]、output_padding[1]、output_padding[2]。	-	-
offset_x	可选属性	默认值为 0，保留字段。	INT	-
y	输出	相当于公式中的($N,C_{out},D_{out},H_{out},W_{out}$)。 数据格式与'x'一致。	FLOAT16、FLOAT32、BFLOAT16、INT8	NCDHW

• 不同groups取值与dtype的组合说明

groups	dtype	x format	filter format	y format
=1	FLOAT16/FLOAT32/BFLOAT16/INT8	NCDHW	NCDHW	NCDHW
=1	FLOAT16/FLOAT32/BFLOAT16/INT8	NCDHW	NDHWC	NCDHW
>1	FLOAT16/FLOAT32/BFLOAT16/INT8	NCDHW	NCDHW	NCDHW
>1	FLOAT16/FLOAT32/BFLOAT16/INT8	NCDHW	NDHWC	NCDHW

约束说明

• x
  • D维度必须为1
• filter
  • D维度必须为1
• strides
  • N和C的维度必须为1。
  • H和W的维度的取值范围必须在[1,63]之间。
  • D维度必须为1
• pads
  • 所有维度的取值范围必须在[0,255]之间。
  • D维度必须为1
• dilations
  • N与C的维度必须为1。
  • W、H维度的取值范围必须在 [1,255] 之间。
  • D维度必须为1
• output_padding
  • N和C维度必须为0，仅允许在深度、高度、宽度方向上添加。