220 Star 945 Fork 693

GVPMindSpore/mindscience

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

sciai.operators.grad

.. py:function:: sciai.operators.grad(net, output_index=0, input_index=-1)

    根据给定的输出索引和输入索引,获取指定网络的导函数网络。所有输出索引将用于求导并结果求和,所有输入索引将分别求导。

    参数:
        - **net** (Cell) - 用于自动微分的网络。
        - **output_index** (int) - 输出索引,从0开始计数。默认值:0。
        - **input_index** (Union(int, tuple[int])) - 需要求导的输入索引,从0开始计数,只允许正向索引。若为-1,则所有指定输入将用于分别求导。默认值:-1。

    输入:
        - **\*inputs** (tuple[Tensor]) - 原网络的输入。

    输出:
        Union(Tensor, tuple[Tensor]),一阶导函数网络的输出。

    异常:
        - **TypeError** - 如果 out_index 不是 int。
        - **TypeError** - 如果 input_index 既不是 `int` 也不是 `int` 的元组/列表。
        - **TypeError** - 如果神经网络的输出既不是 `Tensor` ,也不是 `Tensor` 的元组。
        - **TypeError** - 如果 `input_index` 类型既不是 `int` 也不是 `int` 的元组。
        - **IndexError** - 如果 `input_index` 或 `output_index` 超出范围。
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/mindspore/mindscience.git
git@gitee.com:mindspore/mindscience.git
mindspore
mindscience
mindscience
master

搜索帮助