hzfan opened a new pull request #15911: [Numpy] einsum URL: https://github.com/apache/incubator-mxnet/pull/15911 ## Description ## numpy compatible [einsum](https://docs.scipy.org/doc/numpy/reference/generated/numpy.einsum.html) ### Changes ### - add einsum - add einsum doc - add path optimization - add benchmark - add forward and backward test ## Comments ## - support both explicit and implicit mode - support multiple operands - support broadcast - support 0-dim NDArray and 0-size NDArray - cannot work until multiple tempspace is enabled ## TODO ## - [x] translate einsum_path into c, so that optimization can work in symbol mode - [x] take advantage of tensordot when it is merged - [x] support "greedy" optimization - [ ] support "optimal" optimization Thank @reminisce and @haojin2 for guidance and review.
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services