python学习日记：np.newaxis

import numpy as np

label = np.array([[1,2,3,4],[5,6,7,8]])
print (label.shape)
label = label[np.newaxis, ...]
print (label.shape)

结果：

[[1 2 3 4]
[5 6 7 8]]
(2, 4)
[[[1 2 3 4]
[5 6 7 8]]]
(1, 2, 4)

很明显，增加了一个“无关紧要”的1维，由二维数组变为三维数组了。

用处：caffe中设置label时要求blob是四维的，可以通过这种方式添加。