张量流中的门控层

时间:2019-05-18 16:17:07

标签: tensorflow if-statement neural-network switch-statement

我只有一个输入x,我想根据其值激活不同的神经网络f(x)。目前我有

if x in [-1,-0.5], y = f1(x)
if x in (-0.5,0], y = f2(x)
if x in (0,0.5], y = f3(x)
if x in (0.5,1], y = f4(x)

我用布尔值乘以网络的总和来实现它

y = tf.cast(tf.logical_and(tf.greater(x, -1.), tf.less(x, -.5)), dtype=tf.float32) * f1 + ...

我也想训练选通(即不指定间隔)。哪种结构最适合我的需求?

0 个答案:

没有答案