torch.nn.init
Note
若API“是否支持”为“是”，“限制与说明”为“-”，说明此API和原生API支持度保持一致。
| API名称 | 是否支持 | 限制与说明 |
|---|---|---|
| torch.nn.init.calculate_gain | 是 | - |
| torch.nn.init.uniform_ | 是 | 支持fp32 |
| torch.nn.init.normal_ | 是 | 支持fp32 |
| torch.nn.init.constant_ | 是 | 支持fp32 |
| torch.nn.init.ones_ | 是 | 支持fp32 |
| torch.nn.init.zeros_ | 是 | 支持fp32 |
| torch.nn.init.eye_ | 是 | 支持fp32 |
| torch.nn.init.dirac_ | 是 | 支持fp32 |
| torch.nn.init.xavier_uniform_ | 是 | 支持fp32 |
| torch.nn.init.xavier_normal_ | 是 | 支持fp32 |
| torch.nn.init.kaiming_uniform_ | 是 | 支持fp32 |
| torch.nn.init.kaiming_normal_ | 是 | 支持fp32 |
| torch.nn.init.trunc_normal_ | 是 | - |
| torch.nn.init.orthogonal_ | 是 | 支持fp32 |
| torch.nn.init.sparse_ | 是 | 支持fp32 |