Nensy索引分配的Tensorflow等价物

Question

什么是Tensorflow中的伪（*） - 相当于此？

array[array < 50] = 0 # numpy

我想它应该是这样的：

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

（*）我不假装保持数组可变，既不会被执行，因为我会成为张量。

也许另一种问题是：如何将tf.cond（）的条件张量数组应用于数字数组，这取决于tf.less（）？

Answer 1

你可以做

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

这将返回一个表达式等效于array之后将包含array[array < 50] = 0 。 如果array是TensorFlow变量，则可以使用tf.assign将上述表达式array 。