让量子程序
运行得更好

# 01 · 定义网络与数组
# 用三矩阵链描述张量网络，并准备对应的数值数组。

import numpy as np

# 三矩阵链 A @ B @ C；相同标签表示要收缩的轴
inputs = [("a", "b"), ("b", "c"), ("c", "d")]
output = ("a", "d")  # 结果保留 a、d 两个轴
size_dict = {"a": 2, "b": 3, "c": 4, "d": 2}

# 数组形状对应标签维度，顺序与 inputs 一致
a = np.arange(6, dtype=np.float64).reshape(2, 3)
b = np.arange(12, dtype=np.float64).reshape(3, 4)
c = np.arange(8, dtype=np.float64).reshape(4, 2)
arrays = [a, b, c]