问题描述:

【功能模块】

mindspore 1.6

【操作步骤&问题现象】

1、我在使用mindspore复现pytorch代码时,发现mindspore没有和pytorch的ctx.save_for_backward和ctx.saved_tensors相似的功能

2、我以为可以用self.input和copy()代替,然后发现并不能

3、请问mindspore有事先相似功能的函数吗?没有的话自定义类该怎么写呢?

【截图信息】

pytorch代码:

msp代码

解答:

mindspore不支持上述两种接口

在pynative模式下支持自定义反向,但是计算逻辑跟torch是不一样的https://www.mindspore.cn/docs/programming_guide/zh-CN/r1.6/debug_in_pynative_mode.html

更多推荐