python - 有没有像 einsum 这样可以对称的包？

Question

我有一些张量收缩，例如A[abcd] B[abcd] = C、 whereA[abcd]=A[badc]和B[abcd]=B[badc]。

我检查了einsum https://numpy.org/doc/stable/reference/generated/numpy.einsum.html的手册，但没有找到任何对称选项。

是否有任何选项einsum或其他包可以处理这种对称性，最好是通过 python/julia，可以使收缩更快？

标签： pythonnumpytensor