PyTorch:torch.nn.init

对于给定的非线性函数,返回推荐的增益值。这些值如下所示:nonlinearitygainlinear1conv{1,2,3}d1sigmoid1tanh5/3relusqrt(2)leaky_relusqrt(2/(1+negative_slope^2))参数:nonlinearity – 非线性函数(nn.functional名称)param – 非线性函数的可选参数例子:从均匀分布U(a, b