Python PyTorch uniform_用法及代碼示例

本文簡要介紹python語言中 torch.nn.init.uniform_ 的用法。

用法:

torch.nn.init.uniform_(tensor, a=0.0, b=1.0)

參數：

• tensor-一個n維torch.Tensor

• a-均勻分布的下界

• b-均勻分布的上界

使用從均勻分布 \mathcal{U}(a, b) 中提取的值填充輸入張量。

例子

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