nnf_prelu

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

Provides functionality to define and train neural networks similar to
'PyTorch' by Paszke et al (2019) <doi:10.48550/arXiv.1912.01703> but written entirely in R
using the 'libtorch' library. Also supports low-level tensor operations and
'GPU' acceleration.

Daniel Falbel

torch

Tensors and Neural Networks with 'GPU' Acceleration

Javier Luraschi

Dmitriy Selivanov

Athos Damiani

Christophe Regouby

Krzysztof Joachimiak

Hamada S. Badr

Sebastian Fischer

Maximilian Pichler

 RStudio

nnf_prelu function

<dl><dt>input</dt>
<dd>(N,*) tensor, where * means, any number of additional
dimensions</dd>
<dt>weight</dt>
<dd>(Tensor) the learnable weights</dd></dl>

Arguments

Prelu — nnf_prelu

<dl>

<dt>input</dt>
<dd>(N,*) tensor, where * means, any number of additional
dimensions</dd>


<dt>weight</dt>
<dd>(Tensor) the learnable weights</dd>

</dl>

Prelu

nnf_prelu: Prelu

Description

Usage

Arguments