torch.linalg

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

API名称

是否支持

限制与说明

torch.linalg.norm

支持bf16,fp16,fp32

torch.linalg.matmul

支持fp16,fp32