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.