PyTorch 1.0 中文文档：torch.nn.init

译者：GeneZC

代码语言：javascript复制

torch.nn.init.calculate_gain(nonlinearity, param=None)

返回给定非线性函数的推荐的增益值。对应关系如下表：

参数：

• nonlinearity – 非线性函数 (nn.functional 中的名字)
• param – 对应非线性函数的可选参数

例子

代码语言：javascript复制

>>> gain = nn.init.calculate_gain('leaky_relu')

代码语言：javascript复制

torch.nn.init.uniform_(tensor, a=0, b=1)

用均匀分布

初始化输入 Tensor。

阅读全文／改进本文

tensor 函数

0 人点赞