如何使用MindSpore实现梯度对数据求导retain_graph=True
收藏回复举报
如何使用MindSpore实现梯度对数据求导retain_graph=True
发表于2024-04-22 19:05:54
0 查看

1 系统环境

硬件环境(Ascend/GPU/CPU): Ascend/GPU/CPU

MindSpore版本: 2.2.0

执行模式(PyNative/ Graph): 不限

2 报错信息

2.1 问题描述

假设输入数据x和标签y,通过模型反向传播mindspore.gard获取到了模型的梯度g,现在需要将梯度g进行一定的运算(假设为f(g)),最后f(g)需要对最初的数据x进行求导,如果使用Torch的话,在求梯度g的时候只要 retain_graph=True,MindSpore该如何实现呢?

3 根因分析

******此处由用户填写******

4 解决方案

******此处由用户填写******

包含文字方案和最终脚本代码

请将正确的脚本打包并上传附件

我要发帖子