set_device_sat_mode【免费下载链接】tensorflowAscend TensorFlow Adapter项目地址: https://gitcode.com/cann/tensorflow功能说明设置针对浮点计算的进程级溢出模式。饱和模式计算出现溢出时计算结果会饱和为浮点数极值-MAX。INF/NaN模式遵循IEEE 754标准根据定义输出INF/NaN的计算结果。针对Ascend 950PR/Ascend 950DT支持两种溢出模式饱和模式与INF/NaN模式请保持默认值INF/NaN模式。饱和模式仅用于兼容旧版本后续不再演进且此模式下计算精度可能存在误差。针对Atlas A3 训练系列产品/Atlas A3 推理系列产品支持两种溢出模式饱和模式与INF/NaN模式请保持默认值INF/NaN模式。饱和模式仅用于兼容旧版本后续不再演进且此模式下计算精度可能存在误差。针对Atlas A2 训练系列产品/Atlas A2 推理系列产品支持两种溢出模式饱和模式与INF/NaN模式请保持默认值INF/NaN模式。饱和模式仅用于兼容旧版本后续不再演进且此模式下计算精度可能存在误差。针对Atlas 训练系列产品默认值为“饱和模式”且仅支持“饱和模式”。函数原型def set_device_sat_mode(mode)参数说明参数名输入/输出描述mode输入设置的溢出模式。- 0饱和模式。- 1INF/NaN模式。针对Ascend 950PR/Ascend 950DT请保持默认值“1”。针对Atlas A3 训练系列产品/Atlas A3 推理系列产品请保持默认值“1”。针对Atlas A2 训练系列产品/Atlas A2 推理系列产品请保持默认值“1”。针对Atlas 训练系列产品默认值“0”且仅支持配置为“0”。返回值无约束说明该接口需要在运行时配置网络脚本执行前调用。调用示例以下示例仅针对Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品其他型号处理器开发者无需显式调用此接口。import tensorflow as tf from npu_bridge.npu_init import * ...... # 网络执行时调用如下接口进行溢出模式的设置 npu_plugin.set_device_sat_mode(1) sess.run(xxx)【免费下载链接】tensorflowAscend TensorFlow Adapter项目地址: https://gitcode.com/cann/tensorflow创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考