Not sure if the wrapper you’re talking about is your own custom code, but I real... | Hacker News

Hacker News new | past | comments | ask | show | jobs | submit

login

rsfern 7 months ago | parent | context | favorite | on: Einsum for Tensor Manipulation

Not sure if the wrapper you’re talking about is your own custom code, but I really like using einops lately. It’s got similar axis naming capabilities and it dispatches to both numpy and pytorch

http://einops.rocks/

chpatrick 7 months ago [–]

I just made my own function but that sounds great, I'll check it out!

Guidelines | FAQ | Lists | API | Security | Legal | Apply to YC | Contact