nn_init_calculate_gain

Return the recommended gain value for the given nonlinearity function.

Provides functionality to define and train neural networks similar to
'PyTorch' by Paszke et al (2019) <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

 RStudio

nn_init_calculate_gain function

<dl><dt>nonlinearity</dt>
<dd>the non-linear function</dd>
<dt>param</dt>
<dd>optional parameter for the non-linear function</dd></dl>

Arguments

Calculate gain — nn_init_calculate_gain

<dl>

<dt>nonlinearity</dt>
<dd>the non-linear function</dd>


<dt>param</dt>
<dd>optional parameter for the non-linear function</dd>

</dl>

nn_init_calculate_gain: Calculate gain

Description

Usage

Arguments