（beta）torch_npu.npu.get_amp_supported_dtype

函数原型

功能说明

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

输出说明

List[torch.dtype]

支持的型号

• [object Object]Atlas 训练系列产品[object Object]
• [object Object]Atlas A2 训练系列产品[object Object]
• [object Object]Atlas A3 训练系列产品[object Object]
• [object Object]Atlas 推理系列产品[object Object]