Sequence optimization

Author: JoeyT1994

Project.toml

@@ -1,7 +1,7 @@
 name = "ITensorNetworks"
 uuid = "2919e153-833c-4bdc-8836-1ea460a35fc7"
 authors = ["Matthew Fishman <[email protected]>, Joseph Tindall <[email protected]> and contributors"]
-version = "0.11.27"
+version = "0.11.28"
 
 [deps]
 AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"
@@ -12,34 +12,38 @@ DataStructures = "864edb3b-99cc-5e75-8d2d-829cb0a9cfe8"
 Dictionaries = "85a47980-9c8c-11e8-2b9f-f7ca1fa99fb4"
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
+EinExprs = "b1794770-133b-4de1-afb4-526377e9f4c5"
 Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
 ITensorMPS = "0d1a4710-d33b-49a5-8f18-73bdf49b47e2"
 ITensors = "9136182c-28ba-11e9-034c-db9fb085ebd5"
 IsApprox = "28f27b66-4bd8-47e7-9110-e2746eb8bed7"
 IterTools = "c8e1da08-722c-5040-9ed9-7db0dc04731e"
+JuliaFormatter = "98e50ef6-434e-11e9-1051-2b60c6c9e899"
 KrylovKit = "0b1a1467-8014-51b9-945f-bf0ae24f4b77"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
 NDTensors = "23ae76d9-e61a-49c4-8f12-3f1a16adf9cf"
 NamedGraphs = "678767b0-92e7-4007-89e4-4527a8725b19"
+OMEinsumContractionOrders = "6f22d1fd-8eed-4bb7-9776-e7d684900715"
 PackageExtensionCompat = "65ce6f38-6b18-4e1d-a461-8949797d7930"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
+Revise = "295af30f-e4ad-537b-8983-00126c2a3abe"
 SerializedElementArrays = "d3ce8812-9567-47e9-a7b5-65a6d70a3065"
 SimpleTraits = "699a6c99-e7fa-54fc-8d76-47d257e15c1d"
 SparseArrayKit = "a9a3c162-d163-4c15-8926-b8794fbefed2"
 SplitApplyCombine = "03a91e81-4c3e-53e1-a0a4-9c0c8f19dd66"
+StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 StructWalk = "31cdf514-beb7-4750-89db-dda9d2eb8d3d"
 Suppressor = "fd094767-a336-5f1f-9728-57cf17d0bbfb"
+TensorOperations = "6aa20fa7-93e2-5fca-9bc0-fbd0db3c71a2"
 TimerOutputs = "a759f4b9-e2f1-59dc-863e-4aeb61b1ea8f"
 TupleTools = "9d95972d-f1c8-5527-a6e0-b4b365fa01f6"
 
 [weakdeps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
-EinExprs = "b1794770-133b-4de1-afb4-526377e9f4c5"
 GraphsFlows = "06909019-6f44-4949-96fc-b9d9aaa02889"
-OMEinsumContractionOrders = "6f22d1fd-8eed-4bb7-9776-e7d684900715"
 Observers = "338f10d5-c7f1-4033-a7d1-f9dec39bcaa0"
 
 [extensions]
@@ -66,20 +70,24 @@ ITensorMPS = "0.3"
 ITensors = "0.7, 0.8"
 IsApprox = "0.1, 1, 2"
 IterTools = "1.4.0"
+JuliaFormatter = "1.0.62"
 KrylovKit = "0.6, 0.7, 0.8"
 MacroTools = "0.5"
 NDTensors = "0.3, 0.4"
 NamedGraphs = "0.6.0"
 OMEinsumContractionOrders = "0.8.3, 0.9"
 Observers = "0.2.4"
 PackageExtensionCompat = "1"
+Revise = "3.7.2"
 SerializedElementArrays = "0.1"
 SimpleTraits = "0.9"
 SparseArrayKit = "0.3, 0.4"
 SplitApplyCombine = "1.2"
+StableRNGs = "1.0.2"
 StaticArrays = "1.5.12"
 StructWalk = "0.2"
 Suppressor = "0.2"
+TensorOperations = "5.1.4"
 TimerOutputs = "0.5.22"
 TupleTools = "1.4"
 julia = "1.10"

src/ITensorNetworks.jl

@@ -1,6 +1,7 @@
 module ITensorNetworks
 include("lib/BaseExtensions/src/BaseExtensions.jl")
 include("lib/ITensorsExtensions/src/ITensorsExtensions.jl")
+include("lib/TensorOperationsExt/TensorOperationsExt.jl")
 include("visualize.jl")
 include("graphs.jl")
 include("abstractindsnetwork.jl")

src/contraction_sequences.jl

@@ -1,6 +1,6 @@
 using Graphs: vertices
 using ITensors: ITensor, contract
-using ITensors.ContractionSequenceOptimization: deepmap, optimal_contraction_sequence
+using ITensors.ContractionSequenceOptimization: deepmap
 using ITensors.NDTensors: Algorithm, @Algorithm_str
 using NamedGraphs.Keys: Key
 using NamedGraphs.OrdinalIndexing: th

src/lib/TensorOperationsExt/TensorOperationsExt.jl

@@ -0,0 +1,12 @@
+module TensorOperationsExt
+using ITensors: ITensors, ITensor, dim, inds
+using TensorOperations: optimaltree
+
+function ITensors.optimal_contraction_sequence(tensors::Vector{<:ITensor})
+  network = collect.(inds.(tensors))
+  inds_to_dims = Dict(i => dim(i) for i in unique(reduce(vcat, network)))
+  seq, _ = optimaltree(network, inds_to_dims)
+  return seq
+end
+
+end