如何在Tensorflow中使用張量中的數據？

Question

如您在圖片中看到的，我想在一次訓練中計算總成本，但是tf.equal返回張量類型，而tf.equal(y1[i],y2[i])不能為True 。

我如何使用張量中的數據

Answer 1

您無法在Tensorflow中執行此操作（pyTorch可以執行以下操作）。 原因是TF需要靜態圖。 但是您正在嘗試進行動態評估。 許多人聲稱此靜態圖是TF的缺點。 但實際上，它啟用了許多很酷的功能。 但是，在您使用的情況下，獲得解決方案有點麻煩：

您需要這樣寫：

z = tf.zeros_like(y1)
label_a = z + 2
label_b = z + 20


case_001 = tf.where(tf.equal(y1, z), z + 2, z)
case_002 = tf.where(tf.equal(y2, z), z + 20, z)
switch_op = tf.where(tf.equal(y1, y2), ..., case_001 + case_002)