torch.nn.functional.prelu

torch.nn.functional.prelu(input, weight) → Tensor

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

注意

weight 预计为标量或 1 维张量。如果 weight 是 1 维的，则当 input.dim() >= 2 时，其大小必须与输入通道数（由 input.size(1) 确定）匹配，否则为 1。在 1 维情况下，请注意，当 input 的维度 > 2 时，weight 可以扩展到 input 的形状，但不能使用正常的广播语义。

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