[功能建议] einops 用于数组索引

正如 @colah 在 Twitter 上建议的那样: https://twitter.com/ch402/status/1539774943214178304 > 是否存在一个 "einsum 版本" 的 gather()? 无论如何, gather 是我需要时使用最痛苦的函数(至少自从我开始使用 einsum 以来)。 > 例如, `gather(array, "a b inds[a b] c -> a b c", inds=inds)` 表示使用 inds 或某些东西对第三维进行索引。 Twitter 线程中还有几个语法建议。我并不完全确定这种语法应该是什么样的,但我同意,数组索引语法是 NumPy 中最难理解的部分之一,而 Einops 目前还没有提供相应的支持。