mindquantum.algorithm.qaia.SimCIM
- class mindquantum.algorithm.qaia.SimCIM(J, h=None, x=None, n_iter=1000, batch_size=1, dt=0.01, momentum=0.9, sigma=0.03, pt=6.5)[源代码]
模拟相干伊辛机算法。
参考文献:Annealing by simulating the coherent Ising machine。
说明
为了内存效率,输入数组 'x' 不会被复制,并且会在优化过程中被原地修改。 如果需要保留原始数据,请使用 x.copy() 传入副本。
- 参数:
J (Union[numpy.array, scipy.sparse.spmatrix]) - 耦合矩阵,维度为 \((N \times N)\)。
h (numpy.array) - 外场强度,维度为 \((N, )\)。
x (numpy.array) - 自旋初始化配置,维度为 \((N \times batch\_size)\)。会在优化过程中被修改。如果不提供(
None
),将被初始化为全零数组。默认值:None
。n_iter (int) - 迭代步数。默认值:
1000
。batch_size (int) - 样本个数。默认值为:
1
。dt (float) - 迭代步长。默认值:
1
。momentum (float) - 动量系数。默认值:
0.9
。sigma (float) - 噪声标准差。默认值:
0.03
。pt (float) - 泵浦系数。默认值:
6.5
。
样例:
>>> import numpy as np >>> from mindquantum.algorithm.qaia import SimCIM >>> J = np.array([[0, -1], [-1, 0]]) >>> solver = SimCIM(J, batch_size=5) >>> solver.update() >>> print(solver.calc_cut()) [1. 1. 1. 0. 0.] >>> print(solver.calc_energy()) [-1. -1. -1. 1. 1.]