Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add fit(method="tree") and fix ALS for complex TNs #274

Merged
merged 6 commits into from
Dec 18, 2024
Merged

Conversation

jcmgray
Copy link
Owner

@jcmgray jcmgray commented Dec 14, 2024

This PR adds the function

  • tensor_network_fit_tree for fitting any tree structured TN to any other (not necessarily tree-like) TN. It uses the canonical form of the tree but no intermediates reuse which might be helpful e.g. if the target TN is also a tree.
  • you can access it via tn.fit(target, method="tree")
  • it also fixes the usual ALS fitting for complex TNs
  • it also auto enables lazy tn linear operator iterative solving to allow faster fitting of much larger tensors in the ALS method.

@jcmgray jcmgray changed the title add fit(method="tree") fix ALS for complex TNs add fit(method="tree") and fix ALS for complex TNs Dec 14, 2024
@jcmgray jcmgray merged commit 8e119dc into main Dec 18, 2024
@jcmgray jcmgray deleted the fitting_updates branch December 18, 2024 23:04
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant