Alibaba Group Holding Limited
DISTRIBUTED TENSOR NETWORK CONTRACTION SCHEME WITH SPLITTING BASED ON DYNAMIC ORDERING

Abstract:

Embodiments of the disclosure provide method for performing contraction on a tensor network. The method can include: receiving, by a system, a tensor network comprising a plurality of tensors and a plurality of edges among the plurality of tensors, wherein each edge is associated with a plurality of index elements; determining a contraction order of the tensor network; 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.