pytensor
pytensor copied to clipboard
Add torch conditionals: IfElse
Description
Adds the IfElse node for torch, using torch.where
Related Issue
- [ ] Closes #939
- [ ] Related to #821
Checklist
- [x] Checked that the pre-commit linting/style checks pass
- [x] Included tests that prove the fix is effective or that the new feature works
- [ ] Added necessary documentation (docstrings and/or example notebooks)
- [X] If you are a pro: each commit corresponds to a relevant logical change
Type of change
- [X] New feature / enhancement
- [ ] Bug fix
- [ ] Documentation
- [ ] Maintenance
- [ ] Other (please specify):
@ricardoV94 could you PTAL?
hmmm - this branch got messed up - i'm gonna close the branch and make a new one.