NonZero

NonZero - 13

版本

  • 名称: NonZero (GitHub)

  • : main

  • 起始版本: 13

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符自 版本 13 起可用。

摘要

返回非零元素的索引(按行主序 - 按维度)。NonZero 的行为类似于 numpy.nonzero: https://docs.scipy.org.cn/doc/numpy/reference/generated/numpy.nonzero.html,但对于标量输入,NonZero 生成的输出形状是 (0, N) 而不是 (1, N),这与 Numpy 的行为不同。

输入

  • X (异构) - T

    输入

输出

  • Y (异构) - 张量(int64)

    输出

类型约束

  • T 包含 ( 张量(bfloat16), 张量(bool), 张量(complex128), 张量(complex64), 张量(double), 张量(float), 张量(float16), 张量(int16), 张量(int32), 张量(int64), 张量(int8), 张量(string), 张量(uint16), 张量(uint32), 张量(uint64), 张量(uint8) )

    限制为所有张量类型。

NonZero - 9

版本

  • 名称: NonZero (GitHub)

  • : main

  • 起始版本: 9

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符自 版本 9 起可用。

摘要

返回非零元素的索引(按行主序 - 按维度)。NonZero 的行为类似于 numpy.nonzero: https://docs.scipy.org.cn/doc/numpy/reference/generated/numpy.nonzero.html,但对于标量输入,NonZero 生成的输出形状是 (0, N) 而不是 (1, N),这与 Numpy 的行为不同。

输入

  • X (异构) - T

    输入

输出

  • Y (异构) - 张量(int64)

    输出

类型约束

  • T 包含 ( 张量(bool), 张量(complex128), 张量(complex64), 张量(double), 张量(float), 张量(float16), 张量(int16), 张量(int32), 张量(int64), 张量(int8), 张量(string), 张量(uint16), 张量(uint32), 张量(uint64), 张量(uint8) )

    限制为所有张量类型。