文件最后提交记录最后更新时间
1 个月前
10 天前
1 个月前
8 天前
19 天前
1 个月前
1 个月前
README

NonZeroWithValue

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品
Atlas 训练系列产品 ×

功能说明

  • 算子功能:找出2D输入x的非零元素,静态max-size一次返回valueindexcount三个输出,有效长度由count给出。
  • 计算说明:
    • value:非零元素的值。
    • index:非零元素坐标,坐标主序展平([2, numel],前半段为行号、后半段为列号)。
    • count:非零元素个数。
    • 非零判定遵循IEEE语义:nan、±inf计为非零,-0.0视为零;按行主序遍历,结果确定。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 待计算的目标张量,严格2D。 DOUBLE, FLOAT, FLOAT16, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, BOOL ND
value 输出 非零元素的值,数据类型与x保持一致,静态max-size,shape为[row*col]。 DOUBLE, FLOAT, FLOAT16, INT8, UINT8, INT16, UINT16, INT32, UINT32, INT64, UINT64, BOOL ND
index 输出 非零元素坐标,坐标主序展平,shape为[2*row*col]。 INT32 ND
count 输出 非零元素个数,shape为[1]。 INT32 ND
transpose 属性 标识index排布,仅支持true(坐标主序)。 BOOL -
dtype 属性 控制index输出数据类型,仅支持INT32。 INT -

约束说明

  • 输入x为严格2D,数据类型支持12类:DOUBLE、FLOAT、FLOAT16、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、BOOL;value数据类型与x一致,index/count恒为INT32。
  • 输入x的元素总数(row×col)不超过INT32_MAX(2³¹-1):坐标输出index为INT32,元素总数超限时坐标无法表示,tiling校验拒绝。
  • 输出为静态max-size,有效长度由count给出。
  • 确定性计算:默认确定性实现。

调用说明

调用方式 样例代码 说明
图模式调用 non_zero_with_value_proto.h 通过算子IR(REG_OP(NonZeroWithValue))构图方式调用NonZeroWithValue算子。