nnf_prelu: Prelu

View source: R/nnf-activation.R

nnf_preluR Documentation

Prelu

Description

Applies element-wise the function PReLU(x) = max(0,x) + weight * min(0,x) where weight is a learnable parameter.

Usage

nnf_prelu(input, weight)

Arguments

input

(N,*) tensor, where * means, any number of additional dimensions

weight

(Tensor) the learnable weights


torch documentation built on Aug. 21, 2025, 5:50 p.m.