fa786e5f2a8016725991cf2d02eebe6368dc04b3,dipy/reconst/dti.py,,wls_fit_tensor,#Any#Any#,1154

Before Change

    tol = 1e-6
    data = np.asarray(data)
    data_flat = data.reshape((-1, data.shape[-1]))
    dti_params = np.empty((len(data_flat), 4, 3))

    //obtain OLS fitting matrix
    //U,S,V = np.linalg.svd(design_matrix, False)

After Change

    //ols_fit = np.dot(U, U.T)
    ols_fit = _ols_fit_matrix(design_matrix)

    log_s = np.log(data)
    w = np.exp(np.einsum("...ij,...j", ols_fit, log_s))
    evals, evecs = decompose_tensor(
        from_lower_triangular(
            np.einsum("...ij,...j",
                      pinv_vec(design_matrix * w[..., None]),
                      w * log_s)
        ),
        min_diffusivity=tol / -design_matrix.min()
    )

Instances