R/nnf-activation.R
nnf_prelu.Rd
Applies element-wise the function \(PReLU(x) = max(0,x) + weight * min(0,x)\) where weight is a learnable parameter.
nnf_prelu(input, weight)
(N,*) tensor, where * means, any number of additional dimensions
(Tensor) the learnable weights