我的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更改为所需的结果？
答案 0 :(得分：2)
您可以使用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)