快捷方式

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 (Tensor) – 量化的輸入

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

回傳類型

Tensor

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

取得初學者和進階開發者的深入教學

檢視教學

資源

尋找開發資源並獲得您的問題解答

檢視資源