torch_npu.npu.is_autocast_enabled

API接口

torch_npu.npu.is_autocast_enabled()

功能描述

确认autocast是否可用。

输出说明

bool

示例

>>>torch_npu.npu.is_autocast_enabled()
   True