torch.linalg

Note

若API“是否支持”为“是”,“限制与说明”为“-”,说明此API和原生API支持度保持一致。

API名称 是否支持 限制与说明
torch.linalg.cholesky 支持fp32
torch.linalg.norm 支持bf16,fp16,fp32
torch.linalg.lstsq 可能回退至CPU执行
torch.linalg.matmul 支持fp16,fp32
输入最大支持6维
torch.linalg.ldl_factor -
torch.linalg.qr 支持fp32,fp64,complex64,complex128
torch.linalg.solve_triangular 支持fp32,fp64,complex64,complex128