mxnet-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <>
Subject [GitHub] [incubator-mxnet] wuxun-zhang commented on issue #16833: [Numpy] expands_dims cannot suport scalar
Date Sat, 16 Nov 2019 03:38:33 GMT
wuxun-zhang commented on issue #16833: [Numpy] expands_dims cannot suport scalar
   @stu1130 @reminisce Thanks for reporting this issue. I can reproduce this from my side.
I just figured out the root cause, when input's ndim is 0, it should not execute MKL-DNN pass.
Anyway, I will submit a PR to fix this issue soon. Sorry for any inconvenience. @pengzhao-intel

   Also tested below input, and no error.
   >>> a = np.array([1])
   >>> np.expand_dims(a, 0)
   mkldnn_verbose,info,Intel MKL-DNN v1.0.4 (commit a0a87d662edeef38d01db4ac5dd25f59a1f0881f)
   mkldnn_verbose,info,Detected ISA is Intel AVX-512 with Intel DL Boost
   mkldnn_verbose,exec,cpu,reorder,simple:any,undef,src_f32::blocked:a:f0 dst_f32::blocked:a:f0,num:1,1,11.0342

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:

With regards,
Apache Git Services

View raw message