PadV2【免费下载链接】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 训练系列产品√功能说明算子功能对输入 tensor 进行指定值的常量填充。示例输入 tensor([[0,1,2]]) paddings([[2,2]]) constant_values(0) 输出为([[0,0,0,1,2,0,0]])参数说明参数名输入/输出/属性描述数据类型数据格式x输入待进行填充的原始 tensor。FLOAT、FLOAT16、INT32、INT64、INT16、INT8、UINT8、UINT16、UINT32、UINT64、BOOL、DOUBLE、COMPLEX64、COMPLEX128、BFLOAT16、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN、FLOAT8_E8M0、FLOAT4_E2M1、FLOAT4_E1M2。NDpaddings输入填充配置shape[N, 2]其中 N 为 x 的维度数。INT32、INT64NDconstant_values输入填充常量值标量 tensor。FLOAT、FLOAT16、INT32、INT64、INT16、INT8、UINT8、UINT16、UINT32、UINT64、BOOL、DOUBLE、COMPLEX64、COMPLEX128、BFLOAT16、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN、FLOAT8_E8M0、FLOAT4_E2M1、FLOAT4_E1M2。NDy输出填充后的 tensor。FLOAT、FLOAT16、INT32、INT64、INT16、INT8、UINT8、UINT16、UINT32、UINT64、BOOL、DOUBLE、COMPLEX64、COMPLEX128、BFLOAT16、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN、FLOAT8_E8M0、FLOAT4_E2M1、FLOAT4_E1M2。NDAtlas 训练系列产品 、 Atlas 推理系列产品 、 Atlas 200I/500 A2 推理产品 数据类型不支持BFLOAT16、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN、FLOAT8_E8M0、FLOAT4_E2M1、FLOAT4_E1M2。Atlas A3 训练系列产品/Atlas A3 推理系列产品 、 Atlas A2 训练系列产品/Atlas A2 推理系列产品 数据类型不支持 HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN、FLOAT8_E8M0、FLOAT4_E2M1、FLOAT4_E1M2。输出 shape 计算y.shape[d] x.shape[d] paddings[d][0] paddings[d][1]约束说明维度约束x 的维度数必须在 [1, 8] 范围内paddings 的第一维必须等于 x 的维度数paddings 的第二维必须等于 2数据类型约束x、constant_values、y 必须使用相同的数据类型paddings 必须使用 INT32 或 INT64填充约束负填充slice时输出 shape x.shape[d] left right 0paddings 参数约束paddings 的形状必须为 [rank, 2]其中 rank 为输入 x 的维度数1~8每一行 [left, right] 表示对应维度的填充数量left: 在该维度的开头填充的元素数right: 在该维度的末尾填充的元素数paddings的值可以为正数表示填充负数表示 slice裁剪零表示不填充调用说明调用方式调用样例说明aclnn 调用test_geir_pad_v2.cpp通过 算子IR 接口方式调用 PadV2 算子【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考