torch.linalg

若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