torch.nn.functional.prelu

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

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

注意

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

详见 PReLU。