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。