torch.linalg

API名称

是否支持

限制与说明

torch.linalg.norm

是

支持bf16，fp16，fp32

torch.linalg.matmul

是

支持fp16，fp32
父主题： PyTorch原生API