Function torch::nn::init::calculate_gain¶
Defined in File init.h
Function Documentation¶
-
double
torch::nn::init
::
calculate_gain
(NonlinearityType nonlinearity, double param = 0.01)¶ Return the recommended gain value for the given nonlinearity function.