55d342a0d0e44f3420877a8e9452adba7933ae72,gpflow/kernels/stationaries.py,Stationary,scaled_square_dist,#Stationary#Any#Any#,38

Before Change

        X2 = X2 / self.lengthscale
        X2s = tf.reduce_sum(tf.square(X2), axis=1)
        dist = -2 * tf.linalg.matmul(X, X2, transpose_b=True)
        dist += tf.reshape(Xs, (-1, 1)) + tf.reshape(X2s, (1, -1))
        return dist

    def scaled_euclid_dist(self, X, X2):
        

After Change

        Xs = tf.reduce_sum(tf.square(X), axis=-1)
        X2 = X2 / self.lengthscale
        X2s = tf.reduce_sum(tf.square(X2), axis=-1)
        dist = -2 * tf.tensordot(X, X2, [[-1], [-1]])
        dist += broadcasting_elementwise(tf.add, Xs, X2s)
        return dist

    def scaled_euclid_dist(self, X, X2):
        

Instances