Simplify

Author: JoeyT1994

src/ITensorNetworks.jl

@@ -1,7 +1,6 @@
 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,9 +1,10 @@
 using Graphs: vertices
-using ITensors: ITensor, contract
+using ITensors: ITensor, contract, inds, dim
 using ITensors.ContractionSequenceOptimization: deepmap
 using ITensors.NDTensors: Algorithm, @Algorithm_str
 using NamedGraphs.Keys: Key
 using NamedGraphs.OrdinalIndexing: th
+using TensorOperations: optimaltree
 
 function contraction_sequence(tn::Vector{ITensor}; alg="optimal", kwargs...)
   return contraction_sequence(Algorithm(alg), tn; kwargs...)
@@ -20,3 +21,10 @@ end
 function contraction_sequence(::Algorithm"optimal", tn::Vector{ITensor})
   return optimal_contraction_sequence(tn)
 end
+
+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