其中（）需要1到2個位置參數，但有3個被給出

Question

我有三個數組，X,Y和Z。如果Z的對應元素爲真，我想要放入res和X的元素;否則，我會放入一個來自Y的元素。

我實現這樣的：

X = tf.constant([[1, 2], [3, 4]]) 
Y = tf.constant([[5, 6], [7, 8]]) 
Z = tf.constant([[True, False], [False, True]], tf.bool) 
res = tf.where(Z, X, Y) 
print(res.eval()) 

不過，我得到這個錯誤：

TypeError: where() takes from 1 to 2 positional arguments but 3 were given 

我看着tf.where的definiton從here和我的使用似乎罰款。

任何想法可能是什麼問題？

Answer 1

我懷疑你使用的是舊版本的TensorFlow：

在r0.10 tf.where過去只有2個參數。

tf.where(input, name=None)

https://www.tensorflow.org/versions/r0.10/api_docs/python/math_ops/sequence_comparison_and_indexing#where