Skip to content
This repository was archived by the owner on Sep 11, 2023. It is now read-only.

Use NamedTensors to name dimensions #25

Open
Tracked by #213
JackKelly opened this issue Jun 28, 2021 · 0 comments
Open
Tracked by #213

Use NamedTensors to name dimensions #25

JackKelly opened this issue Jun 28, 2021 · 0 comments
Labels
enhancement New feature or request

Comments

@JackKelly
Copy link
Member

JackKelly commented Jun 28, 2021

https://pytorch.org/docs/stable/named_tensor.html

See this twitter conversation.

In particular, try extending xarray with a little "accessor" extension like dataarray.torch.to_tensor() (as suggested by Joe Hamman on Twitter!)

And see Joe's pull request, which implements xarray-to-pytorch-named-tensors.

But note that torch.stack doesn't yet support named tensors (here's a feature request)

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
enhancement New feature or request
Projects
No open projects
Status: Todo
Development

No branches or pull requests

2 participants