LinearFormNetwork

Author: JoeyT1994

src/ITensorNetworks.jl

@@ -29,6 +29,7 @@ include("caches/abstractbeliefpropagationcache.jl")
 include("caches/beliefpropagationcache.jl")
 include("formnetworks/abstractformnetwork.jl")
 include("formnetworks/bilinearformnetwork.jl")
+include("formnetworks/linearformnetwork.jl")
 include("formnetworks/quadraticformnetwork.jl")
 include("contraction_tree_to_graph.jl")
 include("gauging.jl")

src/abstractitensornetwork.jl

@@ -751,7 +751,7 @@ function split_index(
 end
 
 function inner_network(x::AbstractITensorNetwork, y::AbstractITensorNetwork; kwargs...)
-  return BilinearFormNetwork(x, y; kwargs...)
+  return LinearFormNetwork(x, y; kwargs...)
 end
 
 function inner_network(
@@ -760,12 +760,7 @@ function inner_network(
   return BilinearFormNetwork(A, x, y; kwargs...)
 end
 
-# TODO: We should make this use the QuadraticFormNetwork constructor here. 
-# Parts of the code (tests relying on norm_sqr being two layer and the gauging code
-#  which relies on specific message tensors) currently would break in that case so we need to resolve
-function norm_sqr_network(ψ::AbstractITensorNetwork)
-  return disjoint_union("bra" => dag(prime(ψ; sites=[])), "ket" => ψ)
-end
+norm_sqr_network(ψ::AbstractITensorNetwork) = inner_network(ψ, ψ)
 
 #
 # Printing

src/expect.jl

@@ -26,10 +26,10 @@ function expect(
   (cache!)=nothing,
   update_cache=isnothing(cache!),
   cache_update_kwargs=default_cache_update_kwargs(alg),
-  cache_construction_kwargs=default_cache_construction_kwargs(alg, inner_network(ψ, ψ)),
+  cache_construction_kwargs=default_cache_construction_kwargs(alg, QuadraticFormNetwork(ψ)),
   kwargs...,
 )
-  ψIψ = inner_network(ψ, ψ)
+  ψIψ = QuadraticFormNetwork(ψ)
   if isnothing(cache!)
     cache! = Ref(cache(alg, ψIψ; cache_construction_kwargs...))
   end
@@ -42,7 +42,7 @@ function expect(
 end
 
 function expect(alg::Algorithm"exact", ψ::AbstractITensorNetwork, ops; kwargs...)
-  ψIψ = inner_network(ψ, ψ)
+  ψIψ = QuadraticFormNetwork(ψ)
   return map(op -> expect(ψIψ, op; alg, kwargs...), ops)
 end
 

src/formnetworks/linearformnetwork.jl

@@ -0,0 +1,66 @@
+using ITensors: ITensor, sim
+
+default_dual_link_index_map = sim
+
+struct LinearFormNetwork{
+  V,TensorNetwork<:AbstractITensorNetwork{V},BraVertexSuffix,KetVertexSuffix
+} <: AbstractFormNetwork{V}
+  tensornetwork::TensorNetwork
+  bra_vertex_suffix::BraVertexSuffix
+  ket_vertex_suffix::KetVertexSuffix
+end
+
+function LinearFormNetwork(
+  bra::AbstractITensorNetwork,
+  ket::AbstractITensorNetwork;
+  bra_vertex_suffix=default_bra_vertex_suffix(),
+  ket_vertex_suffix=default_ket_vertex_suffix(),
+  dual_link_index_map=default_dual_link_index_map,
+)
+  bra_mapped = dual_link_index_map(bra; sites=[])
+  tn = disjoint_union(bra_vertex_suffix => dag(bra_mapped), ket_vertex_suffix => ket)
+  return LinearFormNetwork(tn, bra_vertex_suffix, ket_vertex_suffix)
+end
+
+function LinearFormNetwork(blf::BilinearFormNetwork)
+  bra, ket, operator = subgraph(blf, bra_vertices(blf)),
+  subgraph(blf, ket_vertices(blf)),
+  subgraph(blf, operator_vertices(blf))
+  bra_suffix, ket_suffix = bra_vertex_suffix(blf), ket_vertex_suffix(blf)
+  operator = rename_vertices(v -> bra_vertex_map(blf)(v), operator)
+  tn = union(bra, ket, operator)
+  return LinearFormNetwork(tn, bra_suffix, ket_suffix)
+end
+
+bra_vertex_suffix(lf::LinearFormNetwork) = lf.bra_vertex_suffix
+ket_vertex_suffix(lf::LinearFormNetwork) = lf.ket_vertex_suffix
+# TODO: Use `NamedGraphs.GraphsExtensions.parent_graph`.
+tensornetwork(lf::LinearFormNetwork) = lf.tensornetwork
+# TODO: Use `NamedGraphs.GraphsExtensions.parent_graph_type`.
+data_graph_type(::Type{<:LinearFormNetwork}) = data_graph_type(tensornetwork(lf))
+# TODO: Use `NamedGraphs.GraphsExtensions.parent_graph`.
+data_graph(lf::LinearFormNetwork) = data_graph(tensornetwork(lf))
+
+function Base.copy(lf::LinearFormNetwork)
+  return LinearFormNetwork(
+    copy(tensornetwork(lf)), bra_vertex_suffix(lf), ket_vertex_suffix(lf)
+  )
+end
+
+function update(
+  lf::LinearFormNetwork,
+  original_bra_state_vertex,
+  original_ket_state_vertex,
+  bra_state::ITensor,
+  ket_state::ITensor,
+)
+  lf = copy(lf)
+  # TODO: Maybe add a check that it really does preserve the graph.
+  setindex_preserve_graph!(
+    tensornetwork(lf), bra_state, bra_vertex(blf, original_bra_state_vertex)
+  )
+  setindex_preserve_graph!(
+    tensornetwork(lf), ket_state, ket_vertex(blf, original_ket_state_vertex)
+  )
+  return lf
+end

test/test_forms.jl

@@ -5,6 +5,7 @@ using NamedGraphs.NamedGraphGenerators: named_grid
 using ITensorNetworks:
   BeliefPropagationCache,
   BilinearFormNetwork,
+  LinearFormNetwork,
   QuadraticFormNetwork,
   bra_network,
   bra_vertex,
@@ -35,6 +36,10 @@ using Test: @test, @testset
   ψbra = random_tensornetwork(rng, s; link_space=χ)
   A = random_tensornetwork(rng, s_operator; link_space=D)
 
+  lf = LinearFormNetwork(ψbra, ψket)
+  @test nv(lf) == nv(ψket) + nv(ψbra)
+  @test isempty(flatten_siteinds(lf))
+
   blf = BilinearFormNetwork(A, ψbra, ψket)
   @test nv(blf) == nv(ψket) + nv(ψbra) + nv(A)
   @test isempty(flatten_siteinds(blf))
@@ -43,6 +48,9 @@ using Test: @test, @testset
   @test underlying_graph(operator_network(blf)) == underlying_graph(A)
   @test underlying_graph(bra_network(blf)) == underlying_graph(ψbra)
 
+  lf = LinearFormNetwork(blf)
+  @test underlying_graph(ket_network(lf)) == underlying_graph(ψket)
+
   qf = QuadraticFormNetwork(ψket)
   @test nv(qf) == 3 * nv(ψket)
   @test isempty(flatten_siteinds(qf))