阈值ReLU

ThresholdedRelu - 22

版本

  • 名称: ThresholdedRelu (GitHub)

  • 域名: main

  • since_version: 22

  • function: True

  • support_level: SupportType.COMMON

  • 形状推断: True

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

摘要

ThresholdedRelu 接受一个输入数据（Tensor）并生成一个输出数据（Tensor），其中对张量的每个元素应用修正线性函数，即当 x > alpha 时，y = x，否则 y = 0。

函数体

此运算符的函数定义。

<
  domain: "",
  opset_import: ["" : 18]
>
ThresholdedRelu <alpha>(X) => (Y)
{
   Alpha = Constant <value_float: float = @alpha> ()
   AlphaCast = CastLike (Alpha, X)
   Zero = Constant <value: tensor = float {0}> ()
   ZeroCast = CastLike (Zero, X)
   AlphaLessThanX = Less (AlphaCast, X)
   Y = Where (AlphaLessThanX, X, ZeroCast)
}

属性

  • alpha - FLOAT (默认值为 '1.0'):

    阈值

输入

  • X (异构) - T:

    输入张量

输出

  • Y (异构) - T:

    输出张量

类型约束

  • T 在 ( tensor(bfloat16), tensor(double), tensor(float), tensor(float16) ):

    将输入和输出类型限制为浮点张量。

ThresholdedRelu - 10

版本

  • 名称: ThresholdedRelu (GitHub)

  • 域名: main

  • since_version: 10

  • function: True

  • support_level: SupportType.COMMON

  • 形状推断: True

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

摘要

ThresholdedRelu 接受一个输入数据（Tensor）并生成一个输出数据（Tensor），其中对张量的每个元素应用修正线性函数，即当 x > alpha 时，y = x，否则 y = 0。

函数体

此运算符的函数定义。

<
  domain: "",
  opset_import: ["" : 18]
>
ThresholdedRelu <alpha>(X) => (Y)
{
   Alpha = Constant <value_float: float = @alpha> ()
   AlphaCast = CastLike (Alpha, X)
   Zero = Constant <value: tensor = float {0}> ()
   ZeroCast = CastLike (Zero, X)
   AlphaLessThanX = Less (AlphaCast, X)
   Y = Where (AlphaLessThanX, X, ZeroCast)
}

属性

  • alpha - FLOAT (默认值为 '1.0'):

    阈值

输入

  • X (异构) - T:

    输入张量

输出

  • Y (异构) - T:

    输出张量

类型约束

  • T 在 ( tensor(double), tensor(float), tensor(float16) ) 中:

    将输入和输出类型限制为浮点张量。