torch_npu.npu.get_amp_supported_dtype
torch_npu.npu.get_amp_supported_dtype()
获取npu设备支持的数据类型,可能设备支持不止一种数据类型
参数解释:
无参数。
返回值:
List[torch.dtype]
约束条件:
无
示例:
无
父主题:
amp