ITensorList

Author: JoeyT1994

ext/ITensorNetworksEinExprsExt/ITensorNetworksEinExprsExt.jl

@@ -2,10 +2,10 @@ module ITensorNetworksEinExprsExt
 
 using ITensors: Index, ITensor, @Algorithm_str, inds, noncommoninds
 using ITensorNetworks:
-  ITensorNetworks, ITensorNetwork, vertextype, vertex_data, contraction_sequence
+  ITensorNetworks, ITensorList, ITensorNetwork, vertextype, vertex_data, contraction_sequence
 using EinExprs: EinExprs, EinExpr, einexpr, SizedEinExpr
 
-function to_einexpr(ts::Vector{ITensor})
+function to_einexpr(ts::ITensorList)
   IndexType = Any
 
   tensor_exprs = EinExpr{IndexType}[]
@@ -21,7 +21,7 @@ function to_einexpr(ts::Vector{ITensor})
   return SizedEinExpr(sum(tensor_exprs; skip=externalinds_tn), inds_dims)
 end
 
-function tensor_inds_to_vertex(ts::Vector{ITensor})
+function tensor_inds_to_vertex(ts::ITensorList)
   IndexType = Any
   VertexType = Int
 
@@ -36,7 +36,7 @@ function tensor_inds_to_vertex(ts::Vector{ITensor})
 end
 
 function ITensorNetworks.contraction_sequence(
-  ::Algorithm"einexpr", tn::Vector{ITensor}; optimizer=EinExprs.Exhaustive()
+  ::Algorithm"einexpr", tn::ITensorList; optimizer=EinExprs.Exhaustive()
 )
   expr = to_einexpr(tn)
   path = einexpr(optimizer, expr)

ext/ITensorNetworksOMEinsumContractionOrdersExt/ITensorNetworksOMEinsumContractionOrdersExt.jl

@@ -1,6 +1,6 @@
 module ITensorNetworksOMEinsumContractionOrdersExt
 using DocStringExtensions: TYPEDSIGNATURES
-using ITensorNetworks: ITensorNetworks
+using ITensorNetworks: ITensorNetworks, ITensorList
 using ITensors: ITensors, Index, ITensor, inds
 using NDTensors: dim
 using NDTensors.AlgorithmSelection: @Algorithm_str
@@ -9,8 +9,6 @@ using OMEinsumContractionOrders: OMEinsumContractionOrders
 # OMEinsumContractionOrders wrapper for ITensors
 # Slicing is not supported, because it might require extra work to slice an `ITensor` correctly.
 
-const ITensorList = Union{Vector{ITensor},Tuple{Vararg{ITensor}}}
-
 # infer the output tensor labels
 # TODO: Use `symdiff` instead.
 function infer_output(inputs::AbstractVector{<:AbstractVector{<:Index}})

ext/ITensorNetworksTensorOperationsExt/ITensorNetworksTensorOperationsExt.jl

@@ -1,11 +1,11 @@
 module ITensorNetworksTensorOperationsExt
 
 using ITensors: ITensors, ITensor, dim, inds
-using ITensorNetworks: ITensorNetworks
+using ITensorNetworks: ITensorNetworks, ITensorList
 using NDTensors.AlgorithmSelection: @Algorithm_str
 using TensorOperations: TensorOperations, optimaltree
 
-function ITensorNetworks.contraction_sequence(::Algorithm"optimal", tn::Vector{ITensor})
+function ITensorNetworks.contraction_sequence(::Algorithm"optimal", tn::ITensorList)
   network = collect.(inds.(tn))
   inds_to_dims = Dict(i => dim(i) for i in unique(reduce(vcat, network)))
   seq, _ = optimaltree(network, inds_to_dims)

src/contraction_sequences.jl

@@ -4,11 +4,13 @@ using ITensors.NDTensors: Algorithm, @Algorithm_str
 using NamedGraphs.Keys: Key
 using NamedGraphs.OrdinalIndexing: th
 
-function contraction_sequence(tn::Vector{ITensor}; alg="optimal", kwargs...)
+const ITensorList = Union{Vector{ITensor},Tuple{Vararg{ITensor}}}
+
+function contraction_sequence(tn::ITensorList; alg="optimal", kwargs...)
   return contraction_sequence(Algorithm(alg), tn; kwargs...)
 end
 
-function contraction_sequence(alg::Algorithm, tn::Vector{ITensor})
+function contraction_sequence(alg::Algorithm, tn::ITensorList)
   return throw(
     ArgumentError(
       "Algorithm $alg isn't defined for contraction sequence finding. Try loading a backend package like