torch_npu.npu.get_amp_supported_dtype

torch_npu.npu.get_amp_supported_dtype()

获取npu设备支持的数据类型,可能设备支持不止一种数据类型