Tensorflow中的伪(*) - 等效是什么?
array[array < 50] = 0 # numpy
我想它应该是这样的:
array = tf.something(array, ...) # or array2 = ...
# OR
array = array.something(...) # or array2 = ...
(*)我不假装保持数组可变,既不会被执行,也不会因为我是张量而被执行。
也许另一种问题是:如何将tf.cond()的条件张量数组应用于数字数组,这取决于tf.less()?
答案 0 :(得分:5)
你可以做到
tf.select(array < 50, tf.zeros_like(array), array)
将返回与array
之后array[array < 50] = 0
将包含的内容相等的表达式。如果array
是TensorFlow变量,则可以使用tf.assign
将上述表达式分配给array
。