torch.nn.functional.prelu

torch.nn.functional.prelu(input, weight) → 张量

逐元素应用函数 $\text{PReLU}(x) = \max(0,x) + \text{weight} * \min(0,x)$ 其中 weight 是一个可学习的参数。

注意

weight 应为一个标量或一维张量。如果 weight 是一维的， 其大小必须与输入通道的数量匹配，由 input.size(1) 确定，当 input.dim() >= 2 时，否则为1。 在一维情况下，请注意当 input 的维度大于2时，weight 可以通过一种方式扩展 到 input 的形状，这种方式无法使用普通的 广播语义 实现。

有关更多详细信息，请参阅 PReLU。