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.