Skip to content
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

Open
twiecki opened this issue Mar 28, 2021 · 4 comments
Open

Add support for named dimensions and coordinates like in xarray #352

twiecki opened this issue Mar 28, 2021 · 4 comments
Labels
enhancement New feature or request question Further information is requested request discussion

Comments

@twiecki
Copy link
Contributor

twiecki commented Mar 28, 2021

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 of x[0].

Same with dimensions: x.sum(axis="bar"). A lot could probably be borrowed from xarray who are doing this really well.

@twiecki twiecki added the enhancement New feature or request label Mar 28, 2021
@twiecki twiecki changed the title Add named dimensions Add support for named dimensions and coordinates like xarray Apr 5, 2021
@twiecki twiecki changed the title Add support for named dimensions and coordinates like xarray Add support for named dimensions and coordinates like in xarray Apr 5, 2021
@lucianopaz
Copy link
Contributor

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.

@OriolAbril
Copy link
Contributor

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

@jhamman
Copy link

jhamman commented Feb 8, 2023

Cross linking a discussion I just started in the Xarray repo on this subject: pydata/xarray#7515

@brandonwillard
Copy link
Member

Cross linking a discussion I just started in the Xarray repo on this subject: pydata/xarray#7515

Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request question Further information is requested request discussion
Projects
None yet
Development

No branches or pull requests

5 participants