快捷方式

celu

class torch.ao.nn.quantized.functional.celu(input, scale, zero_point, alpha=1.)[原始碼][原始碼]

逐元素應用量化後的 CELU 函式。

CELU(x)=max(0,x)+min(0,α(exp(x/α)1))\text{CELU}(x) = \max(0,x) + \min(0, \alpha * (\exp(x / \alpha) - 1))
引數
  • input (張量) – 量化後的輸入

  • alpha (浮點數) – CELU 公式中的 α\alpha 值。預設值:1.0

返回型別

張量

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源