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.