若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.orthogonal_ |
是 |
支持fp32 |
torch.nn.init.sparse_ |
是 |
支持fp32 |