Module contraction_cost

Expand description

Different methods to compute the computational and memory cost of contraction paths.

Functions§

communication_path_cost: Returns Schroedinger contraction time and space complexity of fully contracting the input tensors assuming all operations occur in parallel.
communication_path_op_costs: Returns Schroedinger contraction time complexity using the critical path metric and using the sum metric. Additionally returns the space complexity.
compute_memory_requirements: Computes the max memory requirements for contracting the tensor network using the given path. Uses memory_estimator to compute the memory required to contract two tensors.
contract_cost_tensors: Returns Schroedinger contraction time complexity of contracting two Tensor objects. Considers cost of complex operations.
contract_op_cost_tensors: Returns Schroedinger contraction time complexity of contracting two Tensor objects. Naive op cost, does not consider costs of multiplication.
contract_path_cost: Returns Schroedinger contraction time and space complexity of fully contracting the input tensors.
contract_size_tensors: Returns Schroedinger contraction space complexity of contracting two Tensor objects.
contract_size_tensors_exact: Returns a rather exact estimate of the memory requirements for contracting tensors i and j.