def Kuu(self, kern, jitter=0.0): Kzz = kern.K(self.Z) Kzz += jitter * tf.eye(len(self), dtype=settings.dtypes.float_type) return Kzz