假设我有张量收缩
A[ab] B[bc] C[cd] -> ABC[ad]
其中,张量{
有两次宫缩
A[ab] B[bc] -> AB[ac], scales n^3
AB[ac] C[cd] -> ABC[ad] scales n^2 m
及
B[bc] C[cd] -> BC[bd] scales n^2 m
A[ab] BC[bd] -> ABC[ad] scales n^2 m
在{
我查过了 https://numpy.org/doc/stable/user/index.htmlhttps://optimized-einsum.readthedocs.io/en/stable/path_finding.html 我不清楚是否考虑了上述情况。我用optimize=“greedy”尝试了几个案例。它们就像第二种模式。我想知道是否考虑了单个张量的维数
目前没有回答
相关问题 更多 >
编程相关推荐