Присвойте значение тензорным полям, которые удовлетворяют условию в Tensorflow

Question

Позвольте мне представить проблему с примером в numpy:

arr  - some numpy array
a, b - constants
arr[arr < a] = b

Теперь я хотел бы реализовать ту же функциональность, но в Tensorflow, где arr будет тензором.Значение тензора неизвестно до времени выполнения.

Как видите, ответ даст разъяснения как для присвоения значений определенным позициям внутри тензоров, так и для выполнения кондиционирования для тензора и извлечения индексов, которые удовлетворяют условию.

Answer 1

Как упомянуто @ xdurch0, tf.where() при надлежащем кондиционировании и трансляции должны добиться цели:

b_broadcast = tf.ones(tf.shape(arr), dtype=arr.dtype) * b
arr = tf.where(tf.less(arr, a), b_broadcast, arr)