torch.prelu

torch.prelu(input, weight) → Tensor

Поэлементно применяет функцию $\text{PReLU}(x) = \max(0,x) + \text{weight} * \min(0,x)$, где weight — обучаемый параметр.

Примечание

Ожидается, что weight является скаляром или одномерным тензором. Если weight является одномерным, его размер должен совпадать с числом входных каналов, определяемым input.size(1), когда input.dim() >= 2, и равным 1 в противном случае. В одномерном случае обратите внимание: когда размерность input больше 2, weight можно расширить до формы input способом, недоступным при использовании обычных правил широковещательного распространения.

Подробнее см. в разделе PReLU.