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