torch.nn.init.uniform_
时间: 2023-04-26 16:01:02 浏览: 373
基于torch.nn的回归、二分类、多分类任务代码
torch.nn.init.uniform_ 是 PyTorch 中的一个初始化函数,用于将张量的值初始化为均匀分布中的随机值。它的用法如下:
torch.nn.init.uniform_(tensor, a=0, b=1)
其中,tensor 是要初始化的张量,a 和 b 是均匀分布的上下界,默认为 0 和 1。该函数会直接修改输入张量的值,并返回该张量。
阅读全文