np.expand_dims
扩展维度,例:
>>> import numpy
>>> a=numpy.arange(10)
>>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> a.shape
(10,)
>>> b=numpy.expand_dims(a,axis=0)
>>> b
array([[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]])
>>> b.shape
(1, 10)
>>> b=numpy.expand_dims(a,axis=1)
>>> b
array([[0],
[1],
[2],
[3],
[4],
[5],
[6],
[7],
[8],
[9]])
>>> b.shape
(10, 1)
>>>