Python PyTorch normal_用法及代码示例

本文简要介绍python语言中 torch.nn.init.normal_ 的用法。

用法:

torch.nn.init.normal_(tensor, mean=0.0, std=1.0)

参数：

• tensor-一个n维torch.Tensor

• mean-正态分布的平均值

• std-正态分布的标准差

使用从正态分布 \mathcal{N}(\text{mean}, \text{std}^2) 中提取的值填充输入张量。

例子

>>> w = torch.empty(3, 5)
>>> nn.init.normal_(w)