CANN算子库Slice切片API

Slice【免费下载链接】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中提取所需的切片。函数原型const aclTensor *Slice(const aclTensor *x, const aclTensor *y, const aclTensor *offset, const aclTensor *size, aclOpExecutor *executor)const aclTensor *Slice(const aclTensor *x, const aclIntArray *offsets, const aclIntArray *size, aclOpExecutor *executor)参数说明参数输入/输出说明x输入输入tensor数据类型支持FLOAT16、FLOAT、BOOL、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、BFLOAT16、UINT64。数据格式支持ND。y输出切片后的输出tensor数据类型支持FLOAT16、FLOAT、BOOL、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、BFLOAT16、UINT64。数据格式支持ND。offsets输入const aclIntArray*类型表示输入x在各个维度切片的起始位置其形状为x的维度。数据类型支持INT32、INT64。数据格式支持ND。offset输入const aclTensor*类型表示输入x在各个维度切片的起始位置其形状为x的维度。数据类型支持INT32、INT64。数据格式支持ND。size输入输入x的各个维度切片的大小其形状为x的维度。支持aclIntArray*、aclTensor*类型。数据类型支持INT32、INT64。数据格式支持ND。executor输入op执行器包含了算子计算流程。说明BFLOAT16仅适用于如下产品Atlas A2 训练系列产品/Atlas A2 推理系列产品Atlas A3 训练系列产品/Atlas A3 推理系列产品返回值说明返回类型和输入tensor一样、shape为size的tensor。约束说明无调用示例// 调用l0op::Slice对每一块进行处理 auto sliceRes l0op::Slice(self, offsetArray, sizeArray, executor); // 调用l0op::Slice对每一块进行处理 auto sliceRes l0op::Slice(xTensor, yTensor, offsetTensor, sizeTensor, executor);【免费下载链接】docs该仓库用于维护cann公共文档项目地址: https://gitcode.com/cann/docs创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考