dev(svd): add support for svd #70
Open
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Collaborator
gausshj
commented
Oct 9, 2025
gausshj
commented
- Add support for svd
- Add test for svd
- Add support for svd - Add test for svd Signed-off-by: Gausshj <[email protected]>
Codecov Report❌ Patch coverage is
📢 Thoughts on this report? Let us know! |
This was referenced Oct 9, 2025
hzhangxyz
requested changes
Oct 9, 2025
grassmann_tensor/tensor.py
Outdated
|
|
||
| tensor = tensor.reshape((left_dim, right_dim)) | ||
|
|
||
| U, S, Vh = torch.linalg.svd(tensor.tensor, full_matrices=full_matrices) |
grassmann_tensor/tensor.py
Outdated
| U, S, Vh = torch.linalg.svd(tensor.tensor, full_matrices=full_matrices) | ||
|
|
||
| k = min(tensor.tensor.shape[0], tensor.tensor.shape[-1]) | ||
| k_index = tensor.tensor.shape.index(k) |
There was a problem hiding this comment.
分别进行svd后，需要允许有cut dimension的操作，这个在tn中很常见。大概就是删掉最小几个singular value，只保留最大的若干个，这个个数使用参数传进来，默认不进行cut，这里两个分块需要分别cut。
- Perform SVD separately on even/odd parity blocks instead of entire
rank-2 tensor
- Update the test case with new implementation
Previously, the SVD was applied directly to the full Grassmann tensor,
which ignored the parity block structure and produced incorrect
decompositions. Now the tensor is split into even/odd blocks before
performing SVD, then recombined via block_diag to ensure correct
parity preservation.
Signed-off-by: Gausshj <[email protected]>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.