a949acef87c6f92589cbd68ff062ecac652539ed,mushroom_rl/solvers/lqr.py,,compute_lqr_Q,#Any#Any#Any#Any#,92

Before Change

    sa = np.hstack((s, a))

    M = _compute_lqr_Q_matrix(lqr, K)
    m = lambda x, y: x.T @ M @ x
    return -1. * pairwise_distances(sa, metric=m).diagonal().reshape((-1, 1))


def compute_lqr_Q_gaussian_policy(s, a, lqr, K, Sigma):
    

After Change

    sa = np.hstack((s, a))

    M = _compute_lqr_Q_matrix(lqr, K)
    return -1. * np.einsum("...k,kl,...l->...", sa, M, sa).reshape(-1, 1)


def compute_lqr_Q_gaussian_policy(s, a, lqr, K, Sigma):
    

Instances