【MindSpore】【自定义求导】mindspore是否支持为一组算子自定义一个求导过程?
问题描述:【操作步骤&问题现象】 如下的pytorch代码的正向过程使用了clamp和round两个算子。其求导过程用一个函数来拟合,请问mindspore中能实现类似的操作吗?我尝试过使用Custom算子实现,但是Custom算子传入的函数似乎不支持使用mindspore中的其他算子,Hybird DSL支持的语法又太少了(不支持if语句?)。【截图信息】 解答:支持的 动态图模式应用 — MindSpore master documentation...