输出非零值索引。
1 2 3 | struct NonzeroParam { uint8_t rsv[8] = {0}; }; |
成员名称 |
类型 |
默认值 |
描述 |
|---|---|---|---|
rsv[8] |
uint8_t |
{0} |
预留参数。 |
参数 |
维度 |
数据类型 |
格式 |
描述 |
|---|---|---|---|---|
x |
[dim_0, dim_1, ..., dim_n] |
int64 |
ND |
输入tensor。 |
参数 |
维度 |
数据类型 |
格式 |
描述 |
|---|---|---|---|---|
output1 |
[inputDimNum, inputNumel] |
int64 |
ND |
非零元素下标。 |
output2 |
[1] |
int64 |
ND |
非零元素个数。 |
仅支持
输入
intensor = {[0, 1, 1, 0, 1]}
输出
outtensor1 = {0, 1, 2, 0, 4}outtensor2 = [3]
输入
intensor = {[0, 1, 1, 0, 1]}
输出
outtensor1 = {0, 1, 2, 0, 4}outtensor2 = [3]