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)