(beta)torch_npu.npu.get_amp_supported_dtype 接口原型torch_npu.npu.get_amp_supported_dtype() 功能描述获取npu设备支持的数据类型,可能设备支持不止一种数据类型。 输出说明List[torch.dtype] 支持的型号Atlas 训练系列产品Atlas A2 训练系列产品Atlas 推理系列产品 父主题: amp