einops

https://github.com/arogozhnikov/einops?tab=readme-ov-file

赞赏