从Tensor中提取切片,从offset位置开始,大小为size。
对应维度offsets,以及offsets + size须在x的对应维度的大小范围内。
struct SliceParam { SVector<int64_t> offsets; SVector<int64_t> size; };
成员名称 |
描述 |
---|---|
offsets |
每个维度切片的起始位置,维度与输入Tensor一致。
|
size |
每个维度切片的大小,维度与输入Tensor一致。
|
参数 |
维度 |
数据类型 |
格式 |
---|---|---|---|
x |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/float/int8/uint8/bool/int32/uint32 |
ND |
参数 |
维度 |
数据类型 |
格式 |
描述 |
---|---|---|---|---|
output |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/float/int8/uint8/bool/int32/uint32 |
ND |
每个维度的大小为size指定的大小。 |