UpsampleNearest2dGrad【免费下载链接】ops-cv本项目是CANN提供的图像处理、目标检测相关的算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-cv产品支持情况产品是否支持Ascend 950PR/Ascend 950DT×Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品×Atlas 推理系列产品×Atlas 训练系列产品×功能说明算子功能UpsampleNearest在exact_mode为false时的反向传播。计算公式$$ gradInput(N, C, H, W) gradOutput( N, C, ceil ( scales_h * H ), ceil ( scales_w * W )) $$参数说明参数名输入/输出/属性描述数据类型数据格式grad_output输入表示反向计算的梯度Tensor对应公式中的gradOutput。FLOAT32、FLOAT16、BFLOAT16NDoutput_size属性表示指定grad_output在H和W维度上的空间大小。size为2且各元素均大于0。LISTINT-input_size属性表示指定grad_input在H和W维度上的空间大小。size为4且最后两个元素均大于零。LISTINT-scales_h可选属性表示输出grad_input的height维度乘数对应公式中的scales_h。默认值为空。FLOAT32-scales_w可选属性表示输出grad_input的width维度乘数对应公式中的scales_w。默认值为空。FLOAT32-grad_input输出表示反向计算的输出张量对应公式中的gradInput。数据类型和数据格式与入参grad_output的数据类型和数据格式保持一致。FLOAT32、FLOAT16、BFLOAT16ND约束说明无调用说明调用方式样例代码说明aclnn接口test_aclnn_upsample_nearest1d_backward通过aclnnUpsampleNearest1dBackward接口方式调用UpsampleNearest2dGrad算子。aclnn接口test_aclnn_upsample_nearest2d_grad通过aclnnUpsampleNearest2dBackward接口方式调用UpsampleNearest2dGrad算子。【免费下载链接】ops-cv本项目是CANN提供的图像处理、目标检测相关的算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-cv创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考