wo() von 1 bis 2 Positionsargumente nimmt, aber 3 wurden gegeben

Question

Ich habe drei Arrays, X, Y und Z. Ich möchte res und Element von X setzen, falls das entsprechende Element von Z wahr ist; Ansonsten werde ich ein Element aus Y setzen.

ich es wie folgt umgesetzt:

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()) 

Allerdings habe ich diese Störung erhalte:

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

ich an der Definiton von tf.where sah von here und meine Nutzung scheint in Ordnung.

Jede Idee, was könnte das Problem sein?

Answer 1

Ich vermute, dass Sie eine alte Version von TensorFlow verwenden:

z.B. in r0.10 tf.where verwendet, um nur 2 Argumente zu nehmen.

tf.where(input, name=None)

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