torch.nn.init

API名称

是否支持

限制与说明

torch.nn.init.calculate_gain

是

  

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_

否