torch.cuda.is_tf32_supported

torch.cuda.is_tf32_supported()[source][source]

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

返回类型:

布尔型