Slice

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品
Atlas 训练系列产品

功能说明

  • 算子功能:从张量中提取一个切片。 此操作是从张量“x”中提取大小为“size”的切片,其中起始位置由“offsets”指定。

y=x[offsets:offsets+size]y = x[offsets:offsets + size]

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入的张量 INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN ND
offsets 输入 slice起始位置 INT32、INT64 ND
size 输入 slice的大小 INT32、INT64 ND
y 输出 输出张量 INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN ND

约束说明

  • 对于输入x的每一个维度n,需要满足:0 <= offsets[i] <= offsets[i] + size[i] <= x_dim[i] for i in [0,n]。

调用说明

调用方式 样例代码 说明
aclnn调用 test_aclnn_slice 通过aclnnSlice方式调用Slice算子