torch.nn.init.uniform(tensor, a=0, b=1)
从均匀分布U(a, b)中生成值,填充输入的张量或变量
参数:
tensor - n维的torch.Tensor
a - 均匀分布的下界
b - 均匀分布的上界
2024-03-18 18:47:16
torch.nn.init.uniform(tensor, a=0, b=1)
从均匀分布U(a, b)中生成值,填充输入的张量或变量
参数:
tensor - n维的torch.Tensor
a - 均匀分布的下界
b - 均匀分布的上界
下一篇:python中的random模块