mlx.nn.prelu#

class prelu(x: array, alpha: array)#

Applies the element-wise parametric ReLU.

$$\text{PReLU}(x)=max(0,x)+a\ast min(0,x)$$

where $a$ is an array.