2.5K Star 8.4K Fork 4.7K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.amp.all_finite.rst 700 Bytes
一键复制 编辑 原始数据 按行查看 历史
俞涵 提交于 2024-04-23 09:58 . modify error html links

mindspore.amp.all_finite

.. py:function:: mindspore.amp.all_finite(inputs)

    检查inputs是否是有效值(无溢出)。

    .. warning::
        这是一个实验性API,后续可能修改或删除。

        此接口只在整网训练情况下用于判断梯度是否溢出,返回结果在不同后端可能存在差异。

    参数:
        - **inputs** (Union(tuple(Tensor), list(Tensor))) - 可迭代的Tensor。

    返回:
        Tensor,布尔类型的标量Tensor。

    教程样例:
        - `自动混合精度 - 损失缩放
          <https://mindspore.cn/tutorials/zh-CN/r2.3.0rc1/advanced/mixed_precision.html#损失缩放>`_
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3.q1

搜索帮助