PyTorch手册PyTorch:torch.nn.init torch.nn.init.calculate_gain(nonlinearity,param=None) 对于给定的非线性函数，返回推荐的增益值。这些值如下所示： nonlinearity gain linear 1 ... AI君 7年前 (2019-03-28) 137℃ 0喜欢