【技术实现步骤摘要】
密度矩阵的计算方法、装置、电子设备及存储介质
[0001]本专利技术实施例涉及信息处理
,尤其涉及一种密度矩阵的计算方法、装置、电子设备及存储介质。
技术介绍
[0002]Gershgorin方法在科学和工程计算中被广泛使用,用于估算矩阵的特征值范围。在量子化学计算中,需要估算正交基组矩阵的特征值范围,用于计算密度矩阵。然而正交基组矩阵的尺寸会达到万级、百万级,计算量巨大。在Gershgorin方法的求解过程中,涉及多个正交基组矩阵元的求和计算过程,在对大规模正交基组矩阵进行求解时,往往需要多个计算节点协同计算,这些计算节点间的数据交互需要满足矩阵元依赖关系,从而需要消耗较长的求解时间。如何能够高效的对正交基组矩阵做特征值范围估算,是量子化学计算中的重要问题。
技术实现思路
[0003]鉴于此,为解决上述技术问题或部分技术问题,本专利技术实施例提供一种密度矩阵的计算方法、装置、电子设备及存储介质。
[0004]第一方面,本专利技术实施例提供一种密度矩阵的计算方法,包括:
[0005]获取 ...
【技术保护点】
【技术特征摘要】
1.一种密度矩阵的计算方法,其特征在于,包括:获取待计算密度矩阵的目标矩阵,并将所述目标矩阵进行分块,得到多个分块矩阵;基于预设规则将每个分块矩阵分配到对应的计算模块中进行数据计算;基于每个计算模块输出的数据计算结果,计算所述目标矩阵的特征值范围;基于所述特征值范围计算所述目标矩阵对应的密度矩阵。2.根据权利要求1所述的方法,其特征在于,所述计算模块包括多个主计算模块和多个副计算模块;所述基于预设规则将每个分块矩阵分配到对应的计算模块中进行数据计算,包括:将位于所述目标矩阵的主对角线上的多个分块矩阵分别分配到对应的主计算模块中;将位于所述目标矩阵的非主对角线上的多个分块矩阵分别分配到对应的副计算模块中。3.根据权利要求2所述的方法,其特征在于,每一个主计算模块和其分配到的分块矩阵所在目标矩阵的同一列的其他分块矩阵对应的多个副计算模块组成一个计算模块组,其中,每个计算模块组中的计算模块串行连接,每个计算模组之间的主计算模块串行连接。4.根据权利要求2所述的方法,其特征在于,所述基于每个计算模块输出的数据计算结果,计算所述目标矩阵的特征值范围,包括:基于所述目标矩阵的第k列元素,由第i个计算模块组通过第一公式和第二公式计算得到多个第一中间值,所述第一公式为:所述第二公式为:基于所述多个第一中间值,通过第三公式和第四公式得到两个第一中间向量,所述第三公式为:X
i
=[x
i1 x
i2 x
i3 x
i4
ꢀ…ꢀ
x
i,n/N
];所述第四公式为:Y
i
=[y
i1 y
i2 y
i3 y
i4
ꢀ…ꢀ
y
i,n/N
]。5.根据权利要求4所述的方法,其特征在于,所述方法还包括:基于所述目标矩阵的第k列元素,由第i个计算模块组的第j个副计算模块通过第五公式计算得到多个第二中间值,所述第五公式为:基于所述多个第二中间值通过第六公式得到全部列组成的第二中间向...
【专利技术属性】
技术研发人员:刘君佩,郭璟,郭振江,唐璇,于莉莉,
申请(专利权)人:上海思朗科技有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。