News for hackers. Stuff that matters.

A comment on Shape Rotation 101: An Intro to Einsum and Jax Transformers.

Re:Shape Rotation 101: An Intro to Einsum and Jax Transformers by cl3misch on Sunday June 23, 2024 @01:40AM (#40765052)
If you're into shape rotations with numpy arrays check out einopt.

Also consider using "None" instead of "np.newaxis". To newcomers it's not as self-explanatory but it results in more readable code imho.

[ Parent | on HN ]
Re:Shape Rotation 101: An Intro to Einsum and Jax Transformers by majidmir on Sunday June 23, 2024 @02:06AM (#40765126)
Or even better einx!
[ Parent | on HN ]
Re:Shape Rotation 101: An Intro to Einsum and Jax Transformers by cl3misch on Sunday June 23, 2024 @03:07AM (#40765339)
Wow, that looks like a whole tensor DSL for numpy (which itself is already an array DSL for Python).
[ Parent | on HN ]
Re:Shape Rotation 101: An Intro to Einsum and Jax Transformers by cycomanic on Sunday June 23, 2024 @02:44AM (#40765250)
Why is ein better than einopt?
[ Parent | on HN ]
Re:Shape Rotation 101: An Intro to Einsum and Jax Transformers by dejavucoder on Sunday June 23, 2024 @02:01AM (#40765109)
thanks, will check out on einops more.
[ Parent | on HN ]