mlx.nn.prelu

class prelu(x: array, alpha: array)

应用逐元素的参数化 ReLU。

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

其中 \(a\) 是一个数组。