mlx.nn.prelu

class mlx.nn.prelu(x: array, alpha: array)

Applies the element-wise function:

\[\text{PReLU}(x) = \max(0,x) + a * \min(0,x)\]

Here \(a\) is an array.