aclnnSlice

产品支持情况

[object Object]Atlas 训练系列产品[object Object]。
[object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]。
[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]。

功能说明

算子功能：在指定维度 $dim$ 上，根据给定的范围 $[start, end]$ 和步长 $step$ ，从输入张量 $self$ 中提取子张量 $out$ 。 $start$ 和 $end$ 可以取 $[0, self.shape[dim]]$ 以外的值，取值后根据以下公式转换为合法值，假设self.shape[dim] = N：

start = \begin{cases} &0, & if\;start < -N \\ &N, & if\;start >= N\\ &(start+N) \% N, & otherwise \end{cases}

end = \begin{cases} &N, & if\; end >= N\\ &start, & else\;if\; (end+N)\%N < start \\ &(end+N)\%N, & otherwise\\ \end{cases}

$out.shape$ 与 $self.shape$ 只有dim轴上不一致，其他轴一致:

out.shape[dim] = ⌊\frac{end - start + step - 1}{step}⌋

函数原型

每个算子分为undefined，必须先调用“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*, 计算输入)：公式中的张量 $self$ ，Device侧的aclTensor。shape支持1到8维，self与out的数据类型一致，支持undefined，undefined支持ND。
  - [object Object]Atlas 训练系列产品[object Object]：数据类型支持FLOAT、FLOAT16、INT32、INT64、INT8、UINT8、BOOL。
  - [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：数据类型支持FLOAT、FLOAT16、INT32、INT64、INT8、UINT8、BOOL、BFLOAT16。
- dim(int64_t, 计算输入)：公式中的参数 $dim$ ，指定的维度，Host侧的整型数值，取值范围为[-self.dim(), self.dim() - 1]。
- start(int64_t, 计算输入)：公式中的参数 $start$ ，切片位置的起始索引，Host侧的整型数值。
- end(int64_t, 计算输入)：公式中的参数 $end$ ，切片位置的终止索引，Host侧的整型数值。
- step(int64_t, 计算输入)：公式中的参数 $step$ ，指定切片的步长，Host侧的整型数值，取值范围为大于0的整数。
- out(aclTensor*, 计算输出)：公式中的张量 $out$ ，Device侧的aclTensor。shape支持1到8维，shape满足计算公式中的推导规则，self与out的数据类型一致，支持undefined，undefined支持ND。
  - [object Object]Atlas 训练系列产品[object Object]：数据类型支持FLOAT、FLOAT16、INT32、INT64、INT8、UINT8、BOOL。
  - [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：数据类型支持FLOAT、FLOAT16、INT32、INT64、INT8、UINT8、BOOL、BFLOAT16。
- workspaceSize(uint64_t*, 出参)：返回需要在Device侧申请的workspace大小。
- executor(aclOpExecutor**, 出参)：返回op执行器，包含了算子计算流程。
返回值：

aclnnStatus：返回状态码，具体参见undefined。

[object Object]

aclnnSlice

参数说明：
- workspace(void*, 入参)：在Device侧申请的workspace内存地址。
- workspaceSize(uint64_t, 入参)：在Device侧申请的workspace大小，由第一段接口aclnnSliceGetWorkspaceSize获取。
- executor(aclOpExecutor*, 入参)：op执行器，包含了算子计算流程。
- stream(aclrtStream, 入参)：指定执行任务的Stream。
返回值：

aclnnStatus：返回状态码，具体参见undefined。

约束说明

无。

调用示例

示例代码如下，仅供参考，具体编译和执行过程请参考undefined。

[object Object]