22edd92d06c201b1f40a656e5bbca84a5f483f2d,gpflow/kernels/misc.py,Coregion,K,#Coregion#Any#Any#,136

Before Change

            X2 = X
        else:
            X2 = tf.cast(X2[:, 0], tf.int32)
        B = tf.linalg.matmul(self.W, self.W, transpose_b=True) + tf.linalg.diag(self.kappa)
        return tf.gather(tf.transpose(tf.gather(B, X2)), X)

    def K_diag(self, X):
        X = tf.cast(X[:, 0], tf.int32)

After Change

        ]
        if X2 is not None:
            shape_constraints.append((X2, [..., "M", 1]))
        tf.debugging.assert_shapes(shape_constraints)

        X = tf.cast(X[..., 0], tf.int32)
        if X2 is None:
            X2 = X

Instances