-
Notifications
You must be signed in to change notification settings - Fork 516
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
Allow expanding and collapsing in aten::view #1082
Conversation
452905f
to
0250a70
Compare
Also @silvasean is there a place I can request write access? This seemed like the best channel to ask in. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for working on this! Adding functionality to this op is very tricky. Below are some preliminary comments. I will do a more detailed pass in a bit
0250a70
to
c872f05
Compare
Added you! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I just have a few small comments
c872f05
to
82f0574
Compare
- Supports cases where the view op expands and collapses dims simulataneously. This does not handle the case where it is neither expanding nor collapsing (e.g. [2, 3] -> [3, 2]) - Additionally fixes a previous bug with adding 1-sized dims on both sides of a tensor with aten.view
82f0574
to
244ba2e
Compare
Thanks for the comments! Let me know if it looks like there are still issues. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!
- Supports cases where the view op expands and collapses dims simulataneously. This does not handle the case where it is neither expanding nor collapsing (e.g. [2, 3] -> [3, 2]) - Additionally fixes a previous bug with adding 1-sized dims on both sides of a tensor with aten.view
* minor edit for the docs/mnist_example Signed-off-by: Alexandre Eichenberger <[email protected]> * additional output Signed-off-by: Alexandre Eichenberger <[email protected]> * responding to suggestions Signed-off-by: Alexandre Eichenberger <[email protected]> * added recommended -O3 option Signed-off-by: Alexandre Eichenberger <[email protected]> Co-authored-by: Tung D. Le <[email protected]>
simultaneously. This does not handle the case where it is neither
expanding nor collapsing (e.g. [2, 3] -> [3, 2])
sides of a tensor with aten.view