GatherOperation
功能
从输入张量中根据索引收集切片，并将这些切片组合成一个新的张量。
图1 GatherOperation
定义
struct GatherParam {
int64_t axis = 0;
int64_t batchDims = 0;
};
成员
成员名称 |
描述 |
|---|---|
axis |
指定要收集切片的轴。默认值为0。 “axis”必须大于或等于0。 |
batchDims |
允许从一个batch的每个元素中收集不同的项目，默认值为0。 “batchDims”必须大于或等于0，且小于或等于axis。 |
输入
参数 |
维度 |
数据类型 |
格式 |
描述 |
|---|---|---|---|---|
x |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/float/bfloat16/int32/uint32 |
ND |
输入tensor。 |
indexs |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
int64/int32/uint32 |
ND |
索引表，值必须在[0, x.shape[axis]]范围内，x与indexs的维数之和小于等于9。indexs的维数必须大于等于“batchdims”。 |
输出
参数 |
维度 |
数据类型 |
格式 |
描述 |
|---|---|---|---|---|
output |
[-1,…,-1] -1表示当前维度的大小没有约束。 |
float16/float/bfloat16/int32/uint32 |
ND |
输出tensor。 |