gokumohandas / fast-weights Goto Github PK

self.A = tf.add(tf.scalar_mul(self.l, self.A),
                         tf.scalar_mul(self.e, tf.batch_matmul(tf.transpose(self.h_s, [0, 2, 1]), self.h_s)))

Traceback (most recent call last):
  File "train.py", line 279, in <module>
    train(FLAGS)
  File "train.py", line 117, in train
    FLAGS.l, FLAGS.e, forward_only=False)
ValueError: too many values to unpack

mu = tf.reduce_mean(self.h_s, reduction_indices=0) # each sample
sigma = tf.sqrt(tf.reduce_mean(tf.square(self.h_s - mu), reduction_indices=0))

mean, var = tf.nn.moments(inputs, [1], keep_dims=True)