mindspore怎么实现pytorch的ctx.save_for_backward和ctx.saved_tensors

问题描述：

【功能模块】

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