np.expand_dims的用法

np.expand_dims（a，axis=？） 即擴展維度，np.expand_dims(a,axis=)即在 a 的相應的axis軸上擴展維度 a = np.array([[1,2],[3,5]]) y = np.expand_dims(a, axis=2) z = np.expand_dims(a, axis=1) print(a.shape) print(y.shape) print(z.

>>阅读原文<<