我在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