torch.nn.init.uniform()的用法

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

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

posted @ 2021-03-15 20:31  吴莫愁258  阅读(3248)  评论(0编辑  收藏  举报