其他分享
首页 > 其他分享> > 关于torch.nn.init.uniform()

关于torch.nn.init.uniform()

作者:互联网

torch.nn.init.uniform(tensor, a=0, b=1)
从均匀分布U(a, b)中生成值,填充输入的张量或变量

参数:
tensor - n维的torch.Tensor
a - 均匀分布的下界
b - 均匀分布的上界

标签:tensor,nn,torch,均匀分布,uniform,init
来源: https://www.cnblogs.com/wmc258/p/14539814.html