Merge remote-tracking branch 'upstream/main' into TensorNetworkFitting

Author: JoeyT1994

.pre-commit-config.yaml

@@ -12,6 +12,6 @@ repos:
     exclude_types: [markdown]  # incompatible with Literate.jl
 
 - repo: "https://github.com/domluna/JuliaFormatter.jl"
-  rev: v2.1.1
+  rev: v2.1.2
   hooks:
   - id: "julia-formatter"

src/formnetworks/bilinearformnetwork.jl

@@ -52,6 +52,12 @@ function Base.copy(blf::BilinearFormNetwork)
   )
 end
 
+function itensor_identity_map(i_pairs::Vector)
+  return prod(i_pairs; init=ITensor(one(Bool))) do i_pair
+    return delta(Bool, dag(first(i_pair)), last(i_pair))
+  end
+end
+
 function BilinearFormNetwork(
   bra::AbstractITensorNetwork,
   ket::AbstractITensorNetwork;
@@ -60,9 +66,13 @@ function BilinearFormNetwork(
 )
   @assert issetequal(flatten_siteinds(bra), flatten_siteinds(ket))
   link_space = isempty(flatten_siteinds(bra)) ? 1 : nothing
-  operator_inds = union_all_inds(siteinds(ket), dual_site_index_map(siteinds(ket)))
-  # TODO: Define and use `identity_network` here.
-  O = ITensorNetwork(Op("I"), operator_inds; link_space)
+  s = siteinds(ket)
+  s_mapped = dual_site_index_map(s)
+  operator_inds = union_all_inds(s, s_mapped)
+
+  O = ITensorNetwork(operator_inds; link_space) do v
+    return inds -> itensor_identity_map(s[v] .=> s_mapped[v])
+  end
   return BilinearFormNetwork(O, bra, ket; dual_site_index_map, kwargs...)
 end
 

src/gauging.jl

@@ -134,7 +134,11 @@ function VidalITensorNetwork(
   if isnothing(cache!)
     cache! = Ref(default_norm_cache(ψ))
   end
-  cache![] = update(cache![]; cache_update_kwargs...)
+
+  if update_cache
+    cache![] = update(cache![]; cache_update_kwargs...)
+  end
+
   return vidalitensornetwork_preserve_cache(ψ; cache=cache![], kwargs...)
 end
 

test/test_contraction_sequence.jl

@@ -1,5 +1,5 @@
 @eval module $(gensym())
-using EinExprs: Exhaustive, Greedy, HyPar
+using EinExprs: Exhaustive, Greedy
 using ITensorNetworks:
   contraction_sequence, norm_sqr_network, random_tensornetwork, siteinds
 using ITensors: ITensors, contract
@@ -49,6 +49,7 @@ using Test: @test, @testset
       Pkg.rm("KaHyPar"; io=devnull)
       res_kahypar_bipartite = contract(tn; sequence=seq_kahypar_bipartite)[]
       @test res_optimal ≈ res_kahypar_bipartite
+      #These tests were leading to CI issues that need to be investigated
       # seq_einexprs_kahypar = contraction_sequence(tn; alg="einexpr", optimizer=HyPar())
       # res_einexprs_kahypar = contract(tn; sequence=seq_einexprs_kahypar)[]
       # @test res_einexprs_kahypar ≈ res_optimal

test/test_expect.jl

@@ -40,10 +40,11 @@ using Test: @test, @testset
   sz_exact = expect(ψ, "Sz"; alg="exact")
   @test sz_bp ≈ sz_exact
 
-  #Test with QNS, product state so should be immediately exact
+  #Test with Quantum Numbers, product state so BP should be exact
   L, χ = 2, 2
   g = named_grid((L, L))
   s = siteinds("S=1/2", g; conserve_qns=true)
+
   ψ = ITensorNetwork(v -> isodd(sum(v)) ? "↑" : "↓", s)
 
   sz_bp = expect(ψ, "Sz"; alg="bp")

test/test_forms.jl

@@ -1,7 +1,7 @@
 @eval module $(gensym())
 using DataGraphs: underlying_graph
 using Graphs: nv
-using NamedGraphs.NamedGraphGenerators: named_grid
+using NamedGraphs.NamedGraphGenerators: named_comb_tree, named_grid
 using ITensorNetworks:
   BeliefPropagationCache,
   BilinearFormNetwork,
@@ -12,16 +12,18 @@ using ITensorNetworks:
   dual_index_map,
   environment,
   flatten_siteinds,
+  inner,
   ket_network,
   ket_vertex,
   operator_network,
   random_tensornetwork,
+  scalar,
   siteinds,
   state_vertices,
   tensornetwork,
   union_all_inds,
   update
-using ITensors: contract, dag, inds, prime, random_itensor
+using ITensors: Index, contract, dag, inds, prime, random_itensor, sim
 using LinearAlgebra: norm
 using StableRNGs: StableRNG
 using TensorOperations: TensorOperations
@@ -84,5 +86,25 @@ using Test: @test, @testset
   ∂qf_∂v_bp = contract(∂qf_∂v_bp)
   ∂qf_∂v_bp /= norm(∂qf_∂v_bp)
   @test ∂qf_∂v_bp ≈ ∂qf_∂v
+
+  #Test having non-uniform number of site indices per vertex
+  g = named_comb_tree((3, 3))
+  s = siteinds("S=1/2", g)
+  s = union_all_inds(s, sim(s))
+  s[(1, 1)] = Index[]
+  s[(3, 3)] = Index[first(s[(3, 3)])]
+  χ = 2
+  rng = StableRNG(1234)
+  ψket = random_tensornetwork(rng, ComplexF64, s; link_space=χ)
+  ψbra = random_tensornetwork(rng, ComplexF64, s; link_space=χ)
+
+  blf = BilinearFormNetwork(ψbra, ψket)
+  @test scalar(blf; alg="exact") ≈ inner(ψbra, ψket; alg="exact")
+
+  lf = LinearFormNetwork(ψbra, ψket)
+  @test scalar(lf; alg="exact") ≈ inner(ψbra, ψket; alg="exact")
+
+  qf = QuadraticFormNetwork(ψket)
+  @test scalar(qf; alg="exact") ≈ inner(ψket, ψket; alg="exact")
 end
 end