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.