我在Python层中编写了一个自定义操作。使用mx.nd.op_name的运算符实现，当输入的形状相同时，它可以正常工作。但是当形状不同时，它会告诉内存不足。自定义操作如下。前进功能完成后，self.output的内存似乎还没有空闲。并且我尝试del self.output，但不起作用。你能提供一些建议吗？
def forward(...):
self.output = mx.nd.op_name(inputs)
do others
def backward(...):
do backward