2016-11-11 40 views
3

我有一個X_train圖像爲:如何重塑4D tensorflow到2D

X-train (37248, 32, 32, 3) 

y_train(37248,43)

我有一個饋字典作爲

train_feed_dict = {features: X_train, labels: train_labels} 
with tf.Session() as session: 
    session.run(init) 
    session.run(loss, feed_dict=train_feed_dict) 

我的特點是:

features = tf.placeholder(tf.float32, shape=[None, 32, 32, 3]) 
features = tf.reshape(features, [-1, 3072]) 

但是當我運行代碼時,我得到t他的錯誤,重塑似乎沒有發生。

ValueError: Cannot feed value of shape (37248, 32, 32, 3) for Tensor 'Reshape_5:0', which has shape '(?, 3072)' 
+1

我認爲你需要跟蹤功能和重新塑造的功能。嘗試使用'features2 = tf.reshape(features,[-1,3072])''。 – drpng

+0

唉,不,同樣的錯誤。 – kevbuntu

+0

進入'feed_dict'的形狀是什麼('features' - 假設'X_train'是你上面描述的形狀)?如果它們是相同的,也許是圖中的處理。 – drpng

回答

2

相信錯誤是二叉:

我不得不將數據拆分成np.array()格式和 flat_features = tf.reshape(功能,[-1,3072])

其特徵= tf.reshape(功能,[-1,3072]),我有兩個問題,所以僅僅通過自身更改名稱牛逼flat_features沒有工作

所以@drpng還就在他的評論中