d5e9eb8ee2987165648a1397e97198e2d20ff5b3,tensorly/decomposition/symmetric_parafac.py,,symmetric_parafac_power_iteration,#,76

Before Change

    if not tl.shape(tensor) == (size, )*order:
        raise ValueError("The input tensor does not have the same size along each mode.")

    factor = tl.zeros((size, rank), **tl.context(tensor))
    weigths = tl.zeros(rank, **tl.context(tensor))

    for i in range(rank):
        eigenval, eigenvec, deflated = symmetric_power_iteration(tensor, n_repeat=n_repeat, n_iteration=n_iteration, verbose=verbose)
        factor = tl.index_update(factor, tl.index[:, i], eigenvec)
        weigths = tl.index_update(weigths, tl.index[i], eigenval)
        tensor = deflated

    return weigths, factor

After Change

    for _ in range(rank):
        eigenval, eigenvec, deflated = symmetric_power_iteration(tensor, n_repeat=n_repeat, n_iteration=n_iteration, verbose=verbose)
        factor.append(eigenvec)
        weigths.append(eigenval)
        tensor = deflated

    factor = tl.stack(factor, axis=1)

Instances