torch.linalg
若API“是否支持”为“是”，“限制与说明”为“-”，说明此API和原生API支持度保持一致。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.linalg.norm |
是 |
支持bf16，fp16，fp32 |
torch.linalg.lstsq |
否 |
- |
torch.linalg.matmul |
是 |
支持fp16，fp32 输入最大支持6维 |
父主题： PyTorch 2.5.1版本
若API“是否支持”为“是”，“限制与说明”为“-”，说明此API和原生API支持度保持一致。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.linalg.norm |
是 |
支持bf16，fp16，fp32 |
torch.linalg.lstsq |
否 |
- |
torch.linalg.matmul |
是 |
支持fp16，fp32 输入最大支持6维 |