DepthToSpace【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2推理产品×Atlas 推理系列产品√Atlas 训练系列产品√功能说明算子功能将深度通道维度的数据移动到空间维度高度和宽度。该算子通过对输入张量的深度通道维度进行重新排列将其转换为空间维度高度和宽度。具体来说它将输入张量的深度维度按照指定的块大小block_size进行划分并将这些深度块重新排列到空间维度中从而增加空间维度的大小同时减少通道维度的深度。计算公式 设输入张量形状为当data_formatNHWC时[batch, height, width, channels]当data_formatNCHW时[batch, channels, height, width]设block_size为B输出张量形状为当data_formatNHWC时[batch, height*B, width*B, channels/(B*B)]当data_formatNCHW时[batch, channels/(B*B), height*B, width*B]其中B*B必须能够整除通道维度。参数说明参数名输入/输出/属性描述数据类型数据格式x输入表示输入张量支持多种数据类型。数据格式必须与data_format属性指定的格式一致FLOAT、FLOAT16、INT8、INT16、INT32、UINT8、UINT16、UINT32、INT64、UINT64、BFLOAT16NHWC、NCHWy输出表示输出张量与输入x具有相同的数据类型。输出形状根据block_size、mode和data_format进行计算与x一致NHWC、NCHWblock_size属性必需表示空间块的尺寸大小必须是大于等于2的整数。通道维度必须能被block_size的平方整除INT-mode属性必需表示数据重排的模式支持DCR和CRD两种模式STRING-data_format属性可选表示数据格式指定输入和输出的维度排列顺序。支持NHWC和NCHW两种格式默认为NHWCSTRING-约束说明输入和输出张量的维度必须是4D输入和输出的数据格式必须相同且仅支持NCHW和NHWC格式block_size必须是大于等于2的整数通道维度必须能被block_size的平方整除mode属性必须是DCR或CRDdata_format属性必须与输入张量的实际格式一致调用说明调用方式样例代码说明图模式test_geir_depth_to_space通过算子IR构图方式调用DepthToSpace算子。【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考