torch.nn.functional.prelu

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

Поэлементно применяет функцию $\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.