压缩

压缩 - 11

版本

  • 名称: Compress (GitHub)

  • : main

  • 自版本: 11

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的算子自版本11起可用。

摘要

沿着给定轴从输入张量中选择条件评估结果为True的切片。如果未提供轴,则在选择元素之前先将输入展平。Compress的行为类似于numpy.compress: https://docs.scipy.org.cn/doc/numpy/reference/generated/numpy.compress.html

属性

  • axis - INT :

    (可选)要获取切片的轴。如果未指定,则在选择元素之前先将输入展平。负值表示从后向前计数维度。接受的范围是[-r, r-1],其中r = rank(input)。

输入

  • input (异构) - T

    秩r >= 1的张量。

  • condition (异构) - T1

    秩1的布尔型张量,用于指示要选择哪些切片或数据元素。它的长度可以小于输入沿轴的长度或如果未指定轴的展平输入的大小。在这种情况下,超过条件长度的数据切片或元素将被丢弃。

输出

  • output (异构) - T

    如果指定了axis,则秩为r的张量。否则,输出是秩为1的张量。

类型约束

  • T in ( tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) )

    将输入和输出类型约束为所有张量类型。

  • T1 in ( tensor(bool) )

    约束为布尔型张量。

压缩 - 9

版本

  • 名称: Compress (GitHub)

  • : main

  • 自版本: 9

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: False

此版本的算子自版本9起可用。

摘要

沿着给定轴从输入张量中选择条件评估结果为True的切片。如果未提供轴,则在选择元素之前先将输入展平。Compress的行为类似于numpy.compress: https://docs.scipy.org.cn/doc/numpy/reference/generated/numpy.compress.html

属性

  • axis - INT :

    (可选)要获取切片的轴。如果未指定,则在选择元素之前先将输入展平。

输入

  • input (异构) - T

    秩r >= 1的张量。

  • condition (异构) - T1

    秩1的布尔型张量,用于指示要选择哪些切片或数据元素。它的长度可以小于输入沿轴的长度或如果未指定轴的展平输入的大小。在这种情况下,超过条件长度的数据切片或元素将被丢弃。

输出

  • output (异构) - T

    如果指定了axis,则秩为r的张量。否则,输出是秩为1的张量。

类型约束

  • T in ( tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) )

    将输入和输出类型约束为所有张量类型。

  • T1 in ( tensor(bool) )

    约束为布尔型张量。