NonzeroOperation

功能

输出非零值索引。

约束

该算子不支持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

数据为非零值的个数。