Rate this Page

Function torch::nn::init::calculate_gain#

Function Documentation#

double torch::nn::init::calculate_gain(NonlinearityType nonlinearity, double param = 0.01)#

Return the recommended gain value for the given nonlinearity function.