-
Notifications
You must be signed in to change notification settings - Fork 54
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
meshgrid implementation #794
Conversation
Codecov Report
@@ Coverage Diff @@
## master #794 +/- ##
==========================================
+ Coverage 95.29% 95.31% +0.01%
==========================================
Files 64 64
Lines 8972 9000 +28
==========================================
+ Hits 8550 8578 +28
Misses 422 422
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
…/heat into feature/775-meshgrid
heat/core/factories.py
Outdated
grids = list(grids) | ||
grids[0], grids[1] = grids[1], grids[0] | ||
|
||
return list(asarray(grid, is_split=splitted) for grid in grids) |
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.
Can we use DNDarray()
here instead of asarray()
?
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.
looks good. only minor changes
Description
Implements meshgrid functionality to Heat. It uses the pytorch function of the same name. That function does not support the indexing parameter. As a hotfix for now, the first and second vector are switched during computation when using 'xy'. If the split is on the first or second axis, they are resplitted beforehand.
Issue/s resolved: #775
Changes proposed:
Type of change
Due Diligence
Does this change modify the behaviour of other functions? If so, which?
no