shithub: opus

Info  •  Files  •  Log  •  Branches

RepeatVector no longer likes tensor input

--- a/dnn/training_tf2/rdovae.py

+++ b/dnn/training_tf2/rdovae.py

@@ -191,7 +191,7 @@

 def var_repeat(x):

-    return RepeatVector(K.shape(x[1])[1])(x[0])

+    return tf.repeat(tf.expand_dims(x[0], 1), K.shape(x[1])[1], axis=1)

 nb_state_dim = 24

-- 

⑨