MXNet 一维数组转成多维数组nd.reshape ----X = X.reshape((1, 1, 6, 8))报错

MXNet 一维数组转成多维数组nd.reshape

X = X.reshape((1, 1, 6, 8))报错

from mxnet import nd
X = nd.ones((6, 8))
X[:, 2:6] = 0
X

output:

[[1. 1. 0. 0. 0. 0. 1. 1.]
 [1. 1. 0. 0. 0. 0. 1. 1.]
 [1. 1. 0. 0. 0. 0. 1. 1.]
 [1. 1. 0. 0. 0. 0. 1. 1.]
 [1. 1. 0. 0. 0. 0. 1. 1.]
 [1. 1. 0. 0. 0. 0. 1. 1.]]
<NDArray 6x8 @cpu(0)>

X = nd.reshape(X,(1,1,6,8))
X

output:

[[[[1. 1. 0. 0. 0. 0. 1. 1.]
   [1. 1. 0. 0. 0. 0. 1. 1.]
   [1. 1. 0. 0. 0. 0. 1. 1.]
   [1. 1. 0. 0. 0. 0. 1. 1.]
   [1. 1. 0. 0. 0. 0. 1. 1.]
   [1. 1. 0. 0. 0. 0. 1. 1.]]]]
<NDArray 1x1x6x8 @cpu(0)>

6x8的二维数组就转化为1x1x6x8的四维数组。其它维数可以照葫芦画瓢。

注意：采用X = X.reshape((1, 1, 6, 8))可能会出错。