如何在TensorFlow中批量计算按元素划分的条件？

Question

我基本上有一批神经元激活在张量A的形状[batch_size, layer_size]中的一个层。让B = tf.square(A)。现在，我想对这个批中每个向量中的每个元素计算以下条件：if abs(e) < 1: e ← 0 else e ← B(e)，其中e是B中与e位于相同位置的元素。我可以用一个tf.cond操作来矢量化整个操作吗？

Answer 1

你可能想看看tf.where(condition, x, y)

关于你的问题：

A = tf.placeholder(tf.float32, [batch_size, layer_size])
B = tf.square(A)

condition = tf.less(tf.abs(A), 1.)

res = tf.where(condition, tf.zeros_like(B), B)