DepthToSpace

产品支持情况

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

功能说明

• 算子功能：将深度（通道）维度的数据移动到空间维度（高度和宽度）。该算子通过对输入张量的深度（通道）维度进行重新排列，将其转换为空间维度（高度和宽度）。具体来说，它将输入张量的深度维度按照指定的块大小（block_size）进行划分，并将这些深度块重新排列到空间维度中，从而增加空间维度的大小，同时减少通道维度的深度。

• 计算公式： 设输入张量形状为：

  • 当data_format="NHWC"时：[batch, height, width, channels]
  • 当data_format="NCHW"时：[batch, channels, height, width]

  设block_size为B，输出张量形状为：

  • 当data_format="NHWC"时：[batch, height*B, width*B, channels/(B*B)]
  • 当data_format="NCHW"时：[batch, channels/(B*B), height*B, width*B]

  其中，B*B必须能够整除通道维度。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	表示输入张量，支持多种数据类型。数据格式必须与data_format属性指定的格式一致	FLOAT、FLOAT16、INT8、INT16、INT32、UINT8、UINT16、UINT32、INT64、UINT64、BFLOAT16	NHWC、NCHW
y	输出	表示输出张量，与输入x具有相同的数据类型。输出形状根据block_size、mode和data_format进行计算	与x一致	NHWC、NCHW
block_size	属性（必需）	表示空间块的尺寸大小，必须是大于等于2的整数。通道维度必须能被block_size的平方整除	INT	-
mode	属性（必需）	表示数据重排的模式，支持"DCR"和"CRD"两种模式	STRING	-
data_format	属性（可选）	表示数据格式，指定输入和输出的维度排列顺序。支持"NHWC"和"NCHW"两种格式，默认为"NHWC"	STRING	-

约束说明

• 输入和输出张量的维度必须是4D
• 输入和输出的数据格式必须相同，且仅支持NCHW和NHWC格式
• block_size必须是大于等于2的整数
• 通道维度必须能被block_size的平方整除
• mode属性必须是"DCR"或"CRD"
• data_format属性必须与输入张量的实际格式一致

调用说明

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