8a4fde49f1bd71bd6d57a31d29c76ba0636997bc,testing/test_conditionals.py,DiagsTest,setUp,#DiagsTest#,13

Before Change

        //the chols are diagonal matrices, with the same entries as the diag representation.
        self.chol = theano.scan(lambda x : tf.diag(x), self.sqrt.T)[0].swapaxes(0,2)

    def test_whiten(self):
        with self.k.tf_mode():

After Change

        //the chols are diagonal matrices, with the same entries as the diag representation.
        self.chol = tf.pack([tf.diag(self.sqrt[:,i]) for i in range(self.num_latent)])
        self.chol = tf.transpose(self.chol, perm=[1,2,0])

    def test_whiten(self):

Instances