若API未标明支持情况则代表该API暂无验证结果,待验证后更新表格。
API名称 |
是否支持 |
限制与说明 |
---|---|---|
torch.linalg.norm |
是 |
支持bf16,fp16,fp32 |
torch.linalg.vector_norm |
||
torch.linalg.matrix_norm |
||
torch.linalg.diagonal |
||
torch.linalg.det |
||
torch.linalg.slogdet |
||
torch.linalg.cond |
||
torch.linalg.matrix_rank |
||
torch.linalg.cholesky |
||
torch.linalg.qr |
||
torch.linalg.lu |
||
torch.linalg.lu_factor |
||
torch.linalg.eig |
||
torch.linalg.eigvals |
||
torch.linalg.eigh |
||
torch.linalg.eigvalsh |
||
torch.linalg.svd |
||
torch.linalg.svdvals |
||
torch.linalg.solve |
||
torch.linalg.solve_triangular |
||
torch.linalg.lu_solve |
||
torch.linalg.lstsq |
||
torch.linalg.inv |
||
torch.linalg.pinv |
||
torch.linalg.matrix_exp |
||
torch.linalg.matrix_power |
||
torch.linalg.cross |
||
torch.linalg.matmul |
是 |
支持fp32,fp16 |
torch.linalg.vecdot |
||
torch.linalg.multi_dot |
||
torch.linalg.householder_product |
||
torch.linalg.tensorinv |
||
torch.linalg.tensorsolve |
||
torch.linalg.vander |
||
torch.linalg.cholesky_ex |
||
torch.linalg.inv_ex |
||
torch.linalg.solve_ex |
||
torch.linalg.lu_factor_ex |
||
torch.linalg.ldl_factor |
||
torch.linalg.ldl_factor_ex |
||
torch.linalg.ldl_solve |