HardSwish

HardSwish - 22

版本

  • 名稱：HardSwish (GitHub)

  • 網域：main

  • 自版本：22

  • 函數：True

  • 支援等級：SupportType.COMMON

  • 形狀推斷：True

此版本的運算子自版本 22 起可用。

摘要

HardSwish 接受一個輸入資料 (張量) 並產生一個輸出資料 (張量)，其中 HardSwish 函數 y = x * max(0, min(1, alpha * x + beta)) = x * HardSigmoid<alpha, beta>(x)，其中 alpha = 1/6 且 beta = 0.5，會逐元素套用至張量。

函數主體

此運算子的函數定義。

<
  domain: "",
  opset_import: ["" : 22]
>
HardSwish (X) => (Y)
{
   HS_X = HardSigmoid <alpha: float = 0.166667, beta: float = 0.5> (X)
   Y = Mul (X, HS_X)
}

輸入

  • X (異質) - T

    輸入張量

輸出

  • Y (異質) - T

    輸出張量

類型約束

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

    將輸入和輸出類型限制為浮點張量。

HardSwish - 14

版本

  • 名稱：HardSwish (GitHub)

  • 網域：main

  • 自版本：14

  • 函數：True

  • 支援等級：SupportType.COMMON

  • 形狀推斷：True

此版本的運算子自版本 14 起可用。

摘要

HardSwish 接受一個輸入資料 (張量) 並產生一個輸出資料 (張量)，其中 HardSwish 函數 y = x * max(0, min(1, alpha * x + beta)) = x * HardSigmoid<alpha, beta>(x)，其中 alpha = 1/6 且 beta = 0.5，會逐元素套用至張量。

函數主體

此運算子的函數定義。

<
  domain: "",
  opset_import: ["" : 14]
>
HardSwish (X) => (Y)
{
   HS_X = HardSigmoid <alpha: float = 0.166667, beta: float = 0.5> (X)
   Y = Mul (X, HS_X)
}

輸入

  • X (異質) - T

    輸入張量

輸出

  • Y (異質) - T

    輸出張量

類型約束

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

    將輸入和輸出類型限制為浮點張量。