通過添加圖像通道來更改形狀

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)