This feature has been merged and will be included as part of the mxnet 2.0. Thank you @mseth10
-- You are receiving this because you are subscribed to this thread. Reply to this email directly or view it on GitHub: https://github.com/apache/incubator-mxnet/issues/17532#issuecomment-667312323
