506b234094009d55d326f1ce957f53f15d4b65ee,geomstats/spd_matrices_space.py,,is_symmetric,#Any#Any#,19

Before Change

    Check if a matrix is symmetric.
    if mat.ndim == 2:
        mat = np.expand_dims(mat, axis=0)
    return np.allclose(mat, mat.transpose(axes=(0, 2, 1)), atol=tolerance)


def make_symmetric(mat):
    Make a matrix fully symmetric to avoid numerical issues.

After Change

    assert mat.ndim == 3
    n_mats, _, _ = mat.shape

    mask = np.zeros(n_mats, dtype=bool)
    for i in range(n_mats):
        mask[i] = np.allclose(mat[i], np.transpose(mat[i]),
                              atol=tolerance)

    return mask

Instances