mindspore.recompute
===================

.. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg
    :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/mindspore/mindspore.recompute.rst
    :alt: 查看源文件


.. py:function:: mindspore.recompute(block, *args, **kwargs)

    该函数用于减少显存的使用，当运行选定的模块时，不再保存其中的前向计算的产生的激活值，我们将在反向传播时，重新计算前向的激活值。

    .. note::
        - 重计算函数只支持继承自Cell对象的模块，
        - 该函数当前只支持PyNative模式，在图模式下，可以尝试使用Cell.recompute()接口，
        - 当使用重计算函数时，传入的网络模块不能使用jit装饰器。

    参数：
        - **block** (Cell) - 需要重计算的网络模块。
        - **args** (tuple) - 指需要重计算的网络模块的前向输入。
        - **kwargs** (dict) - 可选输入。

    返回：
        同block的返回类型相同。

    异常：
        - **TypeError** - 如果 `block` 不是Cell对象。
        - **AssertionError** - 如果执行模式不是PyNative模式。