2017-03-09 253 views
0

我有兩個1D Tensorflow張量,我想將它們按照元素進行比較,並創建一個新的張量記錄它們不同的索引。對於情況下,他們每個店的索引到一個不同的,2D張量,所以如果我可以使用他們喜歡numpy的陣列,我可能會做這樣的事情:比較張量(張量流)

for i in range(0, len(predicted_indices)): 
    if predicted_indices[i] != correct_indices[i]: 
     failed_preds.append(self.input_placeholder[i]) 

但在情況下predicted_indices和correct_indices都是張量,怎麼能我這樣做?

也可以通過其他方式來做到這一點。我試圖存儲我的模型在某個時代出錯的具體示例。

+0

你在尋找'tf.equal(prediction,labels)',它返回一個bool類型張量來指示'prediction'和'labels'之間的區別嗎? –

+0

是的,謝謝! – user2813092

+0

順便說一下,'tensorflow'本身提供了utils函數來衡量包含在'tf.metrics'模塊中的模型的性能,看看它你可能會發現對你有用的東西 –

回答