einops icon indicating copy to clipboard operation
einops copied to clipboard

Published 20 hours ago • verified publisher icon arogozhnikov

Mxnet fails when reducing with large number of dimensions

Open arogozhnikov opened this issue 5 years ago • 0 comments

  • normally, it is an mxnet issue
  • seems it was like for ages (see code around MXNET_SPECIAL_MAX_NDIM)

After digging into mxnet:

  • neighboring reduced axes (and non-reduced axes) are collapsed
  • at most 5 collapsed axes are possible

arogozhnikov avatar Nov 01 '18 09:11 arogozhnikov