| CPC G06F 30/20 (2020.01) [G06F 17/16 (2013.01); G06N 10/00 (2019.01)] | 17 Claims |
|
1. A method for performing contraction on a tensor network, comprising:
receiving, by a system, a tensor network comprising a plurality of tensors and a plurality of edges among the plurality of tensors, wherein the edges comprise one or more open edges and one or more closed edges, and each edge is associated with a plurality of index elements;
determining a contraction order of the tensor network by:
creating a virtual tensor for connecting the one or more open edges;
generating an intermediate tensor network, the intermediate tensor network having the one or more open edges closed by the virtual tensor;
performing tree decomposition on the intermediate tensor network to construct a tree; and
determining the contraction order of the tensor network based on the tree;
determining, among the plurality of edges, one or more edges for generating a plurality of sub-networks based on the tensor network; and
distributing the plurality of sub-networks to a plurality of computing nodes of the system to perform, by the plurality of computing nodes, contraction on the plurality of sub-networks based on the contraction order.
|