This repository was archived by the owner on Nov 17, 2023. It is now read-only.
This repository was archived by the owner on Nov 17, 2023. It is now read-only.
nd.stack causes abort for differently sized arrays #9380
Description
$ pip list | grep mxnet
mxnet (1.0.0.post1)
dca90487275d:~ macaidan$ python
Python 3.6.3 (default, Oct 4 2017, 06:09:15)
[GCC 4.2.1 Compatible Apple LLVM 9.0.0 (clang-900.0.37)] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> from mxnet import nd
>>> a = nd.arange(20).reshape((10, 2))
>>> b = nd.arange(10).reshape((10, 1))
>>> nd.stack(a, b)
[12:09:05] /Users/travis/build/dmlc/mxnet-distro/mxnet-build/dmlc-core/include/dmlc/logging.h:308: [12:09:05] include/mxnet/./tensor_blob.h:276: Check failed: this->shape_.Size() == shape.Size() (10 vs. 20) TBlob.get_with_shape: new and old shape do not match total elements
This causes the Python environment to exit. A simple non-fatal error message would suffice in this case.