mlx.nn.prelu

class prelu(x: array, alpha: array)

Applies the element-wise parametric ReLU.

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

where \(a\) is an array.