コード例 #1
0
ファイル: core.py プロジェクト: wangleiphy/trax
def Selu(x,
         alpha=1.6732632423543772848170429916717,
         lmbda=1.0507009873554804934193349852946):
  return lmbda * np.where(x > 0, x, alpha * np.expm1(x))
コード例 #2
0
ファイル: core.py プロジェクト: wangleiphy/trax
def Elu(x, a=1., **unused_kwargs):
  return np.where(x > 0, x, a * np.expm1(x))