205 Star 842 Fork 618

GVPMindSpore / mindscience

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindflow.operators.batched_jacobian.rst 595 Bytes
一键复制 编辑 原始数据 按行查看 历史
liulei277 提交于 2023-02-24 15:46 . modify all interface

mindflow.operators.batched_jacobian

.. py:function:: mindflow.operators.batched_jacobian(model)

    计算网络模型的雅可比矩阵。

    参数:
        - **model** (mindspore.nn.Cell) - 输入维度为in_channels输出维度为out_channels的网络模型。

    返回:
        Function,用于计算雅可比矩阵的Jacobian实例。输入维度为:[batch_size,in_channels],输出维度为:[out_channels,batch_size,in_channels]。

    .. note::
        要求MindSpore版本 >= 2.0.0调用如下接口: `mindspore.jacrev` 。
1
https://gitee.com/mindspore/mindscience.git
git@gitee.com:mindspore/mindscience.git
mindspore
mindscience
mindscience
master

搜索帮助