-
-
Notifications
You must be signed in to change notification settings - Fork 153
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
Add support for named dimensions and coordinates like in xarray #352
Comments
Just to link this, as it could be useful, Torch does something like this with named tensors. Numpy also has its structured arrays, but those are much more flexible than what we need to get names into dimensions. I think that the key thing that we would need to do as a bare minimum is to handle dimension name propagation through the graph. |
This might be useful to follow and see if making xarray use Aesara as computational backend can male thia possible without extra work on Aesara |
Cross linking a discussion I just started in the Xarray repo on this subject: pydata/xarray#7515 |
Thanks! |
This came up in a conversation with @aseyboldt.
In PyMC3 the
dims
feature that allows naming of shapes leads to great UX. It would be great to push this down into Aesara and extend it. One thing that would be nice is to have named indices so that one could do:x["foo"]
instead ofx[0]
.Same with dimensions:
x.sum(axis="bar")
. A lot could probably be borrowed fromxarray
who are doing this really well.The text was updated successfully, but these errors were encountered: