torch_npu.npu.set_autocast_dtype

torch_npu.npu.set_autocast_dtype(dtype)

设置设备在AMP场景支持的数据类型。

• 参数解释：

  dtype：数据类型。

• 约束条件：

  无

• 示例：

  torch_npu.npu.set_autocast_dtype(torch.float16)