NonzeroOperation

功能

输出非零值索引。

约束

仅在Atlas 800I A2推理产品上支持。

定义

struct NonzeroParam {};

输入

参数

维度

数据类型

格式

描述

x

[-1,…,-1]

-1表示当前维度的大小没有约束。

int64

ND

输入tensor。

输出

参数	维度	数据类型	格式	描述
output1	[inputDimNum, inputNumel]	int64	ND	非零元素下标。
output2	[1]	int64	ND	非零元素个数。

父主题： atb/infer_op_params.h