(a=0.01)
return Fn('LeakyRelu', lambda x: jnp.where(x >= 0, x, a * x))
r"""Returns a ReLU-like layer with linear nonzero outputs for negative inputs.

  .. math::
      f(x) = \left\{ \begin{array}{cl}
          ax & \text{if}\ x \leq 0, \\
          x  & \text{otherwise}.
      \end{array} \right.

  Args:
    a: Slope of line for negative inputs.
r"""Returns a ReLU-like layer with linear nonzero outputs for negative inputs.
r Returns a ReLU - like layer with linear nonzero outputs for negative inputs .
def LeakyRelu(a=0.01):
  r"""Returns a ReLU-like layer with linear nonzero outputs for negative inputs.

  .. math::
      f(x) = \left\{ \begin{array}{cl}
          ax & \text{if}\ x \leq 0, \\
          x  & \text{otherwise}.
      \end{array} \right.

  Args:
    a: Slope of line for negative inputs.
  """
  return Fn('LeakyRelu', lambda x: jnp.where(x >= 0, x, a * x))
def LeakyRelu ( a = 0.01 ) : return Fn ( 'LeakyRelu' , lambda x : jnp . where ( x >= 0 , x , a * x ) )