评价此页

torch.cuda.is_tf32_supported#

torch.cuda.is_tf32_supported()[source]#

返回一个布尔值,指示当前的 CUDA/ROCm 设备是否支持 dtype tf32。

返回类型

布尔值