In the new PyTorch version, there is experimental support for named tensors, which looks like a big deal for example when vectorizing a pipeline or something of the sort. The idea has been floating in the community for a while, I think it will greatly help with axis bugs. What I am not sure is what advantages does it bring compared to, say tsalib?

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

http://nlp.seas.harvard.edu/NamedTensor

https://github.com/ofnote/tsalib