torch_npu.npu.set_autocast_dtype(dtype)
设置设备在AMP场景支持的数据类型。
dtype：数据类型。
torch_npu.npu.set_autocast_dtype(torch.float16)