torch.linalg

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

API名称

是否支持

限制与说明

torch.linalg.norm

支持bf16,fp16,fp32

torch.linalg.lstsq

-

torch.linalg.matmul

支持fp16,fp32

输入最大支持6维