NonzeroOperation

Hardware Model	Supported
Atlas 350 accelerator cards	x
Atlas A3 inference products/Atlas A3 training products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas training products	x
Atlas inference products	x
Atlas 200I/500 A2 inference products	x

Outputs a non-zero index.

struct NonzeroParam {
    uint8_t rsv[8] = {0};
};

Member	Type	Default Value	Description
rsv[8]	uint8_t	{0}	Reserved

Parameter	Dimension	Data Type	Format	Description
x	[dim_0, dim_1, ..., dim_n]	int64	ND	Input tensor.

Parameter	Dimension	Data Type	Format	Description
output1	[inputDimNum, inputNumel]	int64	ND	Subscript of a non-zero element. inputDimNum is the number of dimensions (n+1) of the input tensor x. inputNumel is the product of all dimensions (dim_0 * dim_1 ... dim_n) of the input tensor x.
output2	[1]	int64	ND	Number of non-zero elements.

Parameter

Dimension

Data Type

Format

Description

output1

[inputDimNum, inputNumel]

int64

Subscript of a non-zero element.

inputDimNum is the number of dimensions (n+1) of the input tensor x.

inputNumel is the product of all dimensions (dim_0 * dim_1 *... * dim_n) of the input tensor x.

output2

[1]

int64

Number of non-zero elements.

One-dimensional scenario

Input

intensor = {[0, 1, 1, 0, 1]}

Output

outtensor1 = {1, 2, 4, 0, 0}
outtensor2 = [3]

Two-dimensional scenario

Input

intensor = {[[0, 1, 1, 0, 1], [0, 1, 1, 0, 1]]}

Output

outtensor1 = {[[0, 0, 0, 1, 1, 1, 0, 0, 0, 0], [1, 2, 4, 1, 2, 4, 0, 0, 0, 0]]} 
outtensor2 = [6]

Parent topic: atb/infer_op_params.h