torch.linalg
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.linalg.norm |
是 |
支持bf16，fp16，fp32 |
torch.linalg.cholesky |
否 |
|
torch.linalg.matmul |
是 |
支持fp16，fp32 |
父主题： PyTorch原生API
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.linalg.norm |
是 |
支持bf16，fp16，fp32 |
torch.linalg.cholesky |
否 |
|
torch.linalg.matmul |
是 |
支持fp16，fp32 |