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