a1d15164eceb07b8c98e95df75e540467524a384,gpflow/likelihoods.py,SwitchedLikelihood,_partition_and_stitch,#SwitchedLikelihood#Any#Any#,477

Before Change

        Y = args[-1]
        ind = tf.gather(tf.transpose(Y), tf.shape(Y)[1]-1)  // ind = Y[:,-1]
        ind = tf.cast(ind, tf.int32)
        Y = tf.transpose(tf.gather(tf.transpose(Y), tf.range(0, tf.shape(Y)[1]-1)))  // Y = Y[:,:-1]
        args[-1] = Y

        // split up the arguments into chunks corresponding to the relevant likelihoods

After Change

        Y = args[-1]
        ind = Y[:, -1]
        ind = tf.cast(ind, tf.int32)
        Y = Y[:, :-1]
        args[-1] = Y

        // split up the arguments into chunks corresponding to the relevant likelihoods
        args = zip(*[tf.dynamic_partition(X, ind, self.num_likelihoods) for X in args])

Instances