与pytorch中c++的einsum源码一致
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
处理流程与pytorch中c++的einsum源码一致,在ipynb文件里,可用通过输出每一步的结果,查看einsum算子是如何工作的。