np.expand_dims函数

即扩展维度，np.expand_dims(a,axis=)即在相应的axis轴上扩展维度
a = np.array([[1,2],[3,5]])

b=(a==0).astype(np.float)

y = np.expand_dims(a, axis=2)
z = np.expand_dims(a, axis=1)
print(a.shape)
print(y.shape)
print(z.shape)
输出
(2, 2)
(2, 2, 1)
(2, 1, 2)

作者：ClarenceHoo
链接：https://www.jianshu.com/p/213bb17c71ca
来源：简书
简书著作权归作者所有，任何形式的转载都请联系作者获得授权并注明出处。