f01a7761d70d727ef4ea42c05464c2f85679ecd1,gpflow/kernels/base.py,Kernel,slice_cov,#Kernel#Any#,79

Before Change

        N = cov.shape[0]
        I = len(act_dims)
        cov_reshaped = tf.reshape(cov, (-1, cov.shape[-1], cov.shape[-1]))
        return tf.reshape(cov_reshaped[..., act_dims, act_dims], (N, I, I))

    @abc.abstractmethod

After Change

        cov_reshaped = tf.reshape(cov, [-1, nlast, nlast])
        gather1 = tf.gather(tf.transpose(cov_reshaped, [2, 1, 0]), dims)
        gather2 = tf.gather(tf.transpose(gather1, [1, 0, 2]), dims)
        cov = tf.reshape(tf.transpose(gather2, [2, 0, 1]),
                         tf.concat([cov_shape[:-2], [ndims, ndims]], 0))

        // nbatch = cov.shape[0]
        // cov_reshaped = tf.reshape(cov, (-1, nlast, nlast))
        // return tf.reshape(cov_reshaped[..., dims, dims], (nbatch, ndims, ndims))

Instances