torch.nn.init中常用的函数:
- torch.nn.init.uniform_
- torch.nn.init.normal_
- torch.nn.init.constant_
- torch.nn.init.ones_
- torch.nn.init.zeros_
- torch.nn.init.eye_
- torch.nn.init.sparse_
torch.nn.init.uniform_(tensor, a=0.0, b=1.0)
按均匀分布 U ( a , b ) U(a,b) U
torch.nn.init中常用的函数:
torch.nn.init.uniform_(tensor, a=0.0, b=1.0)
按均匀分布 U ( a , b ) U(a,b) U
微信扫码登录