acc533ac66fa4d19420a36006ece53a440d03761,gpflow/covariances/kuus.py,,_Kuu,#,15

Before Change

    Zmu, Zlen = kern.slice(feat.Z(), feat.scales())
    idlengthscales2 = tf.square(kern.lengthscales() + Zlen)
    sc = tf.sqrt(
        tf.expand_dims(idlengthscales2, 0) + tf.expand_dims(idlengthscales2, 1) - tf.square(
            kern.lengthscales()))
    d = feat._cust_square_dist(Zmu, Zmu, sc)
    Kzz = kern.variance() * tf.exp(-d / 2) * tf.reduce_prod(kern.lengthscales() / sc, 2)

After Change

def _Kuu(feat: Multiscale, kern: RBF, *, jitter=0.0):
    Zmu, Zlen = kern.slice(feat.Z(), feat.scales())
    idlengthscales2 = tf.square(kern.lengthscales() + Zlen)
    sc = tf.sqrt(idlengthscales2[None, ...] + idlengthscales2[:, None, ...]
                 - kern.lengthscales() ** 2)
    d = feat._cust_square_dist(Zmu, Zmu, sc)
    Kzz = kern.variance() * tf.exp(-d / 2) * tf.reduce_prod(kern.lengthscales() / sc, 2)
    Kzz += jitter * tf.eye(len(feat), dtype=Kzz.dtype)

Instances