pytensor icon indicating copy to clipboard operation
pytensor copied to clipboard

Add torch conditionals: IfElse

Open Ch0ronomato opened this issue 1 year ago • 1 comments

Description

Adds the IfElse node for torch, using torch.where

Related Issue

  • [ ] Closes #939
  • [ ] Related to #821

Checklist

Type of change

  • [X] New feature / enhancement
  • [ ] Bug fix
  • [ ] Documentation
  • [ ] Maintenance
  • [ ] Other (please specify):

Ch0ronomato avatar Jul 17 '24 19:07 Ch0ronomato

@ricardoV94 could you PTAL?

Ch0ronomato avatar Jul 29 '24 22:07 Ch0ronomato

hmmm - this branch got messed up - i'm gonna close the branch and make a new one.

Ch0ronomato avatar Aug 15 '24 18:08 Ch0ronomato