StridedSlice

产品支持情况

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

功能说明

  • 算子功能:按照指定的起始、结束位置和步长,从输入张量中提取一个子张量。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入的张量 INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN ND
begin 输入 每个维度的起始值 INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN ND
end 输入 每个维度的结束值 INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN ND
strides 输入 每个维度上每个点取值的跨度 INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN ND
begin_mask 属性 这个值指定哪些维度的begin被忽略 INT32、INT64 ND
end_mask 属性 这个值指定哪些维度的end被忽略 INT32、INT64 ND
ellipsis_mask 属性 从指定的维度开始全选,直到遇到用户指定begin才退出 INT32、INT64 ND
new_axis_mask 属性 在指定位上增加维度为1的shape INT32、INT64 ND
shrink_axis_mask 属性 把对应索引处维度强制降为1 INT32、INT64 ND
y 输出 输出张量 INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、FLOAT、FLOAT16、BF16、BOOL、COMPLEX32、COMPLEX64、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN ND

约束说明

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_strided_slice 通过aclnnStridedSlice接口方式调用StridedSlice算子。
图模式调用 test_strided_slice 通过算子IR构图方式调用StridedSlice算子