python - tensorflow : Update Certain Tensor Indices On a Condition

Question

我正在使用 Tensorflow，我想将以下条件应用于我的张量:

if(value > 0):
   value = log(value)
else:
   value = -log(-value)

目前，我正在做以下事情。

minone = tf.constant(-1, dtype=tf.float32)
condition = tf.less(X, minone)
idx = tf.where(condition)
portion = tf.gather_nd(X, idx)
log = -tf.log(-portion)
X = tf.scatter_nd_update(X, idx, log)

但是，这只在 X 类型为 tf.Variable 而不是普通 Tensor 类型时才有效。

如何实现Tensor类型变量的工作条件？

Answer 1

value = tf.sign(value) * tf.log(tf.abs(value))