Slice
支持的产品型号
Atlas 训练系列产品
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 推理系列产品
函数功能
从输入tensor中提取想要的切片。
函数原型
- const aclTensor *Slice(const aclTensor *x, const aclTensor *offset, const aclTensor *size, aclOpExecutor *executor)
- const aclTensor *Slice(const aclTensor *x, const aclTensor *y, const aclTensor *offset, const aclTensor *size, aclOpExecutor *executor)
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
x |
输入 |
被切片的tensor，数据类型支持FLOAT16、FLOAT、BOOL、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、BFLOAT16、UINT64。数据格式支持ND。 说明：
其中BFLOAT16适用于如下产品型号： Atlas A2训练系列产品/Atlas 800I A2推理产品 |
y |
输出 |
切片产生的tensor，数据类型支持FLOAT16、FLOAT、BOOL、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、BFLOAT16、UINT64。 说明：
其中BFLOAT16适用于如下产品型号： Atlas A2训练系列产品/Atlas 800I A2推理产品 |
offset |
输入 |
输入x的各个维度切片的起始位置，其形状为x的维度。数据类型支持INT32、INT64。数据格式支持ND。 |
size |
输入 |
输入x的各个维度切片的大小，其形状为x的维度。数据类型支持INT32、INT64。数据格式支持ND。 |
executor |
输入 |
op执行器，包含了算子计算流程。 |
返回值说明
返回类型和输入tensor一样、shape为size的tensor。
约束说明
无
父主题： 基础开发接口