2.4K Star 8.1K Fork 4.3K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.amp.all_finite.rst 697 Bytes
一键复制 编辑 原始数据 按行查看 历史
宦晓玲 提交于 2024-04-18 09:38 . modify the error 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/master/advanced/mixed_precision.html#损失缩放>`_
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助

53164aa7 5694891 3bd8fe86 5694891