torch.nn.functional.prelu

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

Applies element-wise the function $\text{PReLU}(x) = \max(0,x) + \text{weight} * \min(0,x)$ where weight is a learnable parameter.

See PReLU for more details.