@sandeep-krishnamurthy I am not sure whether this should be implemented in the backend. It needs to be based on several other operators, as it would be complicated and inefficient to directly forward and backward as a whole (especially backward). Could you point to me where I should work if I need to use other operators? Or, alternatively, is there a way of storing temporary data in forward for use in backward? Thanks!
[ Full content available at: https://github.com/apache/incubator-mxnet/issues/10840 ] This message was relayed via gitbox.apache.org for [email protected]
