GenIndex#
General#
The GenIndex operation creates an index tensor along a specified axis of an input tensor. The resulting index tensor has the same shape as the input tensor, with each element representing the index along the specified axis.
Operation Attributes#
Attribute Name |
Description |
Value Type |
Supported Values |
Required or Optional |
|---|---|---|---|---|
Specifies the dimension along which index values are generated. |
s64 |
An s64 value in the range of [-r, r-1] where r = rank(src) |
Required |
Execution Arguments#
Input#
Index |
Argument Name |
Required or Optional |
|---|---|---|
0 |
|
Required |
Output#
Index |
Argument Name |
Required or Optional |
|---|---|---|
0 |
|
Required |
Supported Data Types#
The GenIndex operation supports the following data type combinations.
Src |
Dst |
|---|---|
f32 |
s32 |
bf16 |
s32 |
f16 |
s32 |