Tensor de flujo equivalente para la asignación indexada de Numpy

¿Qué es el pseudo (*) – equivalente en Tensorflow para esto?

array[array < 50] = 0 # numpy 

Supongo que debería ser algo como:

 array = tf.something(array, ...) # or array2 = ... # OR array = array.something(...) # or array2 = ... 

(*) No pretendo mantener la matriz mutable ni que se ejecute en este momento ya que sería un tensor.

Quizás otra forma de preguntar esto es: ¿Cómo sería el código para aplicar una matriz de tensores condicionales de tf.cond () dependiendo de tf.less () a una matriz de números?

Tu puedes hacer

 tf.select(array < 50, tf.zeros_like(array), array) 

que devolverá una expresión equivalente a la que contendrá la array[array < 50] = 0 después de la array[array < 50] = 0 . Si array era una variable TensorFlow, puede usar tf.assign para asignar la expresión anterior a array .