5 ms·
Also see Einops: https://github.com/arogozhnikov/einops https://github.com/arogozhnikov/einops, which uses a einsum-like notation for various tensor operations
by teddykoker 4y ago
Also see Einops: https://github.com/arogozhnikov/einops https://github.com/arogozhnikov/einops, which uses a einsum-like notation for various tensor operations used in deep learning.
https://einops.rocks/pytorch-examples.html https://einops.rocks/pytorch-examples.html shows how it can be used to implement various neural network architectures in a more simplified manor.
- PartiallyTyped 4y agoIn addition, einsum in jax uses einops.
- pizza 4y agoSomebody also realized that much of the time you can use one single function to describe all 3 of the einops operations. I present to you, einop: https://github.com/cgarciae/einop https://github.com/cgarciae/einop
- johnnymellor 4y agoEinops looks nice! It reminds me of https://github.com/deepmind/einshape https://github.com/deepmind/einshape which is another attempt at unifying reshape, squeeze, expand_dims, transpose, tile, flatten, etc under an einsum-inspired DSL.