输出非零值索引。
struct NonzeroParam {};
参数 |
维度 |
数据类型 |
格式 |
描述 |
|---|---|---|---|---|
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]