Nensy索引分配的Tensorflow等价物

时间:2016-02-22 19:52:09

标签: python numpy tensorflow

Tensorflow中的伪(*) - 等效是什么?

array[array < 50] = 0 # numpy

我想它应该是这样的:

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

(*)我不假装保持数组可变,既不会被执行,也不会因为我是张量而被执行。

也许另一种问题是:如何将tf.cond()的条件张量数组应用于数字数组,这取决于tf.less()?

1 个答案:

答案 0 :(得分:5)

你可以做到

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

将返回与array之后array[array < 50] = 0将包含的内容相等的表达式。如果array是TensorFlow变量,则可以使用tf.assign将上述表达式分配给array