torch_npu.npu.set_autocast_dtype API接口torch_npu.npu.set_autocast_dtype(dtype) 功能描述设置设备在AMP场景支持的数据类型。 参数说明dtype：数据类型。 示例torch_npu.npu.set_autocast_dtype(torch.float16) 父主题： amp
