[ 参数完全一致 ] torch.nn.init.calculate_gain¶
torch.nn.init.calculate_gain¶
torch.nn.init.calculate_gain(nonlinearity, param=None)
paddle.nn.initializer.calculate_gain¶
paddle.nn.initializer.calculate_gain(nonlinearity, param=None)
两者参数和用法完全一致，具体如下：
参数映射¶
| PyTorch | PaddlePaddle | 备注 |
|---|---|---|
| nonlinearity | nonlinearity | 非线性激活函数的名称，两者参数和用法完全一致。 |
| param | param | 某些激活函数的参数，默认为 None，两者参数和用法完全一致。 |