NonzeroOperation（代码开放）
产品支持情况
硬件型号 |
是否支持 |
|---|---|
√ |
|
√ |
|
x |
|
x |
|
x |
功能说明
输出非零值索引。
定义
1 2 3 | struct NonzeroParam { uint8_t rsv[8] = {0}; }; |
参数列表
成员名称 |
类型 |
默认值 |
描述 |
|---|---|---|---|
rsv[8] |
uint8_t |
{0} |
预留参数。 |
输入
参数 |
维度 |
数据类型 |
格式 |
描述 |
|---|---|---|---|---|
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]