张量收缩

[www.ziwang.com]

https://github.com/jcmgray/cotengra
cotengra是一个 Python 库，用于收缩涉及大量张量的张量网络或 Einsum 表达式 - 主要文档可以在 cotengra.readthedocs.io 中找到。其中一些关键功能包括：cotengra

一个明确的收缩树对象，可以灵活地构建、修改和可视化
一个“超级优化器”，在调整生成元参数的同时对树进行采样
动态切片可节省大量内存和并行性
支持超边张量网络，从而支持任意方程
可以提供给numpy.einsum，opt_einsum，quimb等的路径
通过 autoray 对来自许多库的张量执行收缩