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)