nn_init_calculate_gain {torch}R Documentation

Calculate gain

Description

Return the recommended gain value for the given nonlinearity function.

Usage

nn_init_calculate_gain(nonlinearity, param = NULL)

Arguments

nonlinearity

the non-linear function

param

optional parameter for the non-linear function


[Package torch version 0.13.0 Index]