aclnnSlice
支持的产品型号
- Atlas 训练系列产品。
- Atlas A2训练系列产品/Atlas 800I A2推理产品。
接口原型
每个算子分为，必须先调用“aclnnSliceGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnSlice”接口执行计算。
aclnnStatus aclnnSliceGetWorkspaceSize(const aclTensor *self, int64_t dim, int64_t start, int64_t end, int64_t step, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)aclnnStatus aclnnSlice(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
功能描述
算子功能：在指定维度上，根据给定的范围和步长，从输入张量中提取子张量。
aclnnSliceGetWorkspaceSize
参数说明：
self(aclTensor*, 计算输入)：Device侧的aclTensor，数据类型支持FLOAT、FLOAT16、INT32、INT64、INT8、UINT8、BOOL、BFLOAT16（仅Atlas A2训练系列产品/Atlas 800I A2推理产品支持）。支持，支持ND。
dim(int64_t, 计算输入)：指定的维度，Host侧的整型数值，数据类型为INT32、INT64。
start(int64_t, 计算输入)：切片位置的起始索引，Host侧的整型数值，数据类型为INT32、INT64。
end(int64_t, 计算输入)：切片位置的终止索引，Host侧的整型数值，数据类型为INT32、INT64。
step(int64_t, 计算输入)：指定切片的步长，Host侧的整型数值，数据类型为INT32、INT64。
out(aclTensor*, 计算输出)：Device侧的aclTensor，数据类型支持FLOAT、FLOAT16、INT32、INT64、INT8、UINT8、BOOL、BFLOAT16（仅Atlas A2训练系列产品/Atlas 800I A2推理产品支持）。支持，支持ND。
workspaceSize(uint64_t*, 出参)：返回用户需要在Device侧申请的workspace大小。
executor(aclOpExecutor**, 出参)：返回op执行器，包含了算子计算流程。
返回值：
aclnnSlice
参数说明：
workspace(void*, 入参)：在Device侧申请的workspace内存地址。
workspaceSize(uint64_t, 入参)：在Device侧申请的workspace大小，由第一段接口aclnnSliceGetWorkspaceSize获取。
executor(aclOpExecutor*, 入参)：op执行器，包含了算子计算流程。
stream(aclrtStream, 入参)：指定执行任务的 AscendCL Stream流。
返回值：
约束与限制
无。