输出非零值索引。
该算子不支持Atlas 推理系列产品(Ascend 310P AI处理器)。
struct NonzeroParam {};
参数 |
维度 |
数据类型 |
格式 |
描述 |
---|---|---|---|---|
input |
[-1,…,-1] |
int64 |
ND |
输入tensor。 |
参数 |
维度 |
数据类型 |
格式 |
描述 |
---|---|---|---|---|
output1 |
[inputDimNum, inputNumel] |
int64 |
ND |
输出tensor。inputDimNum为input的维度数。inputNumel为input的总大小。 |
output2 |
[1] |
int64 |
ND |
数据为非零值的个数。 |