mlx.nn.PReLU

class PReLU(num_parameters=1, init=0.25)

应用逐元素的参数化 ReLU。

应用逐元素的 \(\max(0, x) + a * \min(0, x)\)，其中 \(a\) 是一个数组。

有关函数式等效项，请参阅 prelu()。

参数:

• num_parameters – 要学习的 \(a\) 的数量。默认值：1

• init – \(a\) 的初始值。默认值：0.25

方法