torch.nn.functional.prelu

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

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

注意

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

更多详情请见 PReLU 。