我基本上有一批神经元激活在张量A
的形状[batch_size, layer_size]
中的一个层。让B = tf.square(A)
。现在,我想对这个批中每个向量中的每个元素计算以下条件:if abs(e) < 1: e ← 0 else e ← B(e)
,其中e
是B
中与e
位于相同位置的元素。我可以用一个tf.cond
操作来矢量化整个操作吗?
发布于 2016-06-19 14:17:20
你可能想看看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)
https://stackoverflow.com/questions/37912161
复制相似问题