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

[Feature] Add batch dimension to input states for time-dependent solvers #294

Merged
merged 4 commits into from
Nov 15, 2024

Conversation

vytautas-a
Copy link
Collaborator

@vytautas-a vytautas-a commented Nov 15, 2024

Add batch dimension to input states for time-dependent solvers. Also introduce flag in config of time-dependent solvers to choose sparse/dense tensors for calculations.

Closes #292 and #274

@vytautas-a vytautas-a added the feature New feature or request label Nov 15, 2024
@vytautas-a vytautas-a self-assigned this Nov 15, 2024
@vytautas-a vytautas-a changed the title Add batch dimension to input states for time-dependent solvers [Feature] Add batch dimension to input states for time-dependent solvers Nov 15, 2024
Copy link
Collaborator

@chMoussa chMoussa left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hey @vytautas-a thanks for looking into this. Just having a few comments regarding a few lines

pyqtorch/time_dependent/solvers.py Outdated Show resolved Hide resolved
tests/test_analog.py Show resolved Hide resolved
tests/test_analog.py Outdated Show resolved Hide resolved
@vytautas-a vytautas-a merged commit a11da37 into main Nov 15, 2024
9 checks passed
@vytautas-a vytautas-a deleted the va/sparse/dense-flag-timedep branch November 15, 2024 13:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[Bug] Mesolve checks for initial state
3 participants