画像チャンネルを追加して形状を変更してください

Question

私のx_train.shapeは[50, 12, 6]です。その後、[50,12, 6, 3]に変更しようとしました。 「3」は画像チャンネル用です。

numpy expand_dim(np.expand_dim(x_train, axis=3)は、[50,12,6,1]に変更されました。

x_train.shapeを希望の結果に変更するにはどうすればよいですか？

Answer 1

を使用できnp.repeat()

a = a[..., None] # equivalent to np.expand_dims(a, axis=3) 
a = a.repeat(repeats=3, axis=3) 
print a.shape 
# (50, 12, 6, 3)