torch.nn.init

若API未标明支持情况,则代表该API的支持情况待验证。

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_

是