220 Star 945 Fork 693

GVPMindSpore/mindscience

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
sciai.operators.JacobianWeights.rst 714 Bytes
一键复制 编辑 原始数据 按行查看 历史
MaZhiming 提交于 2023-09-10 18:18 +08:00 . [SciAI] init sciai docs

sciai.operators.JacobianWeights

.. py:class:: sciai.operators.JacobianWeights(model, out_shape, out_type=ms.float32)

    关于权重的雅可比矩阵。输入元组中的最后一个输入Tensor是权重Parameter,其余为网络的输入。

    参数:
        - **model** (Cell) - 用于计算关于权重的雅可比的网络。
        - **out_shape** (tuple) - 网络输出的形状。
        - **out_type** (type) - Mindspore 数据类型。 默认值:ms.float32。

    输入:
        - **\*x** (tuple[Tensor]) - 网络的输入,和用于求雅可比矩阵的权重。

    输出:
        Tensor,网络输出关于指定权重的雅可比矩阵。
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/mindspore/mindscience.git
git@gitee.com:mindspore/mindscience.git
mindspore
mindscience
mindscience
master

搜索帮助