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))
可能会出错。