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.
