CANN/ops-math PadV3Grad算子
PadV3Grad
【免费下载链接】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 训练系列产品 | √ |
功能说明
算子功能：
根据不同条件调用不同的底层padgrad算子（pad_v3_grad_replicate/pad_v3_grad_replication/pad_v4_grad/reflection_pad3d_grad）。
调用说明
| 算子名称 | 对应README | 调用条件 |
|---|---|---|
| circular_pad_grad | circular_pad_grad | 当输入模式（mode）“circular”时调用算子circular_pad_grad。 |
| pad_v4_grad | pad_v4_grad | 当输入张量维度小于等于4，输入模式（mode）为“reflect”时调用算子pad_v4_grad。 |
| pad_v3_grad_replicate | pad_v3_grad_replicate | 当输入张量维度小于等于4，输入模式（mode）为“edge”时调用算子pad_v3_grad_replicate。 |
| pad_v3_grad_replication | pad_v3_grad_replication | 当输入张量维度等于5，输入模式（mode）为“edge”时调用算子pad_v3_grad_replication。 |
| reflection_pad3d_grad | reflection_pad3d_grad | 当输入张量维度等于5，输入模式（mode）为“reflect”时调用算子reflection_pad3d_grad。 |
【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考