| nnf_prelu {torch} | R 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 |
[Package torch version 0.13.0 Index]