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]
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]。
调用说明