be14d17696f15d11a8075e4943c718320ca97a45,gpflow/conditionals/mo_sample_conditionals.py,,_sample_conditional,#Any#Any#Any#Any#,13

Before Change

    g_mu, g_var = independent_cond(Xnew, feat, kern, f, white=white, q_sqrt=q_sqrt,
                                   full_output_cov=False, full_cov=False)  // [N, L], [N, L]
    g_sample = sample_mvn(g_mu, g_var, "diag")  // [N, L]
    f_sample = tf.einsum("pl, nl -> np", kern.W(), g_sample)
    return f_sample

After Change

                                  white=white, q_sqrt=q_sqrt)  // [..., N, L], [..., N, L]
    g_sample = sample_mvn(g_mu, g_var, "diag", num_samples=num_samples)  // [..., (S), N, L]
    f_mu, f_var = mix_latent_gp(kern.W, g_mu, g_var, full_cov, full_output_cov)
    f_sample = tf.tensordot(g_sample, kern.W, [[-1], [-1]])  // [..., N, P]
    return f_sample, f_mu, f_var

Instances