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.dirac_()

• torch.nn.init.xavier_uniform_()

• torch.nn.init.xavier_normal_()

• torch.nn.init.kaiming_uniform_()

• torch.nn.init.kaiming_normal_()

• torch.nn.init.trunc_normal_()

• torch.nn.init.orthogonal_()

• torch.nn.init.sparse_()