EnSUM或OptInSUM优化路径考虑张量的维数吗?

2024-09-28 03:25:16 发布

您现在位置:Python中文网/ 问答频道 /正文

假设我有张量收缩 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

在{}假设下,第二次收缩更快

我的问题是,EnSUM的搜索,贪婪还是最优,考虑张量的维度,例如,上面的例子?

我查过了 https://numpy.org/doc/stable/user/index.htmlhttps://optimized-einsum.readthedocs.io/en/stable/path_finding.html 我不清楚是否考虑了上述情况。我用optimize=“greedy”尝试了几个案例。它们就像第二种模式。我想知道是否考虑了单个张量的维数


Tags: httpsabhtmlcdbdadac例子

热门问题