华为CANN算子库Pad接口

📅 2026/7/5 18:43:13
华为CANN算子库Pad接口
Pad【免费下载链接】docs该仓库用于维护cann公共文档项目地址: https://gitcode.com/cann/docs产品支持情况Ascend 950PR/Ascend 950DT 不支持Atlas A3 训练系列产品/Atlas A3 推理系列产品 支持Atlas A2 训练系列产品/Atlas A2 推理系列产品 支持Atlas 200I/500 A2 推理产品 不支持Atlas 推理系列产品 支持Atlas 训练系列产品 支持功能说明将输入tensor按照paddings的大小对各个维度进行填充填充值为0。函数原型const aclTensor* Pad(const aclTensor* self, const aclTensor* paddings, aclOpExecutor* executor)参数说明参数输入/输出说明self输入待填充的输入tensor数据类型支持FLOAT16、FLOAT、INT16、UINT16、INT32、INT64、BFLOAT16、INT8。数据格式支持ND。paddings输入输入tensor每个维度被填充的大小形状为[self.dim, 2]数据类型支持INT32、INT64。数据格式支持ND。executor输入op执行器包含了算子计算流程。说明BFLOAT16和INT8仅适用于如下产品Atlas A2 训练系列产品/Atlas A2 推理系列产品Atlas A3 训练系列产品/Atlas A3 推理系列产品返回值说明返回被填充了0的tensor。约束说明无调用示例//调用l0op::Pad对self进行补维 l0op::Pad(self, paddingsTensor, executor);【免费下载链接】docs该仓库用于维护cann公共文档项目地址: https://gitcode.com/cann/docs创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考