torch.linalg

Note

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

API名称 是否支持 限制与说明
torch.linalg.norm 支持bf16,fp16,fp32
torch.linalg.lstsq -
torch.linalg.matmul 支持fp16,fp32
输入最大支持6维
torch.linalg.ldl_factor -