SpaceToDepth

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	√
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必须能够整除高度和宽度维度。

参数说明

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

约束说明

无

调用说明

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