This thread is very timely, as it relates to issues I've been trying to solve with my coding of the last week. I think it simplest just to present the code and detailed PR description as the clearest form of my thinking. Happy to discuss further. Please see:
https://github.com/apache/incubator-mxnet/pull/8313 ----------------------------------------------------------------------------------- This email message is for the sole use of the intended recipient(s) and may contain confidential information. Any unauthorized review, use, disclosure or distribution is prohibited. If you are not the intended recipient, please contact the sender by reply email and destroy all copies of the original message. -----------------------------------------------------------------------------------