3 ms·
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 di
by rsfern 2y ago
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/ http://einops.rocks/
- chpatrick 2y agoI just made my own function but that sounds great, I'll check it out!