wuxun-zhang commented on issue #16833: [Numpy] expands_dims cannot suport scalar
URL: https://github.com/apache/incubator-mxnet/issues/16833#issuecomment-554600286
@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
array([[1.]])
>>>
```
----------------------------------------------------------------
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:
users@infra.apache.org
With regards,
Apache Git Services
|