Add files via upload

Author: chubbc

ABCD1.png

@@ -0,0 +1,8 @@
+# This file is machine-generated - editing it directly is not advised
+
+[[Libdl]]
+uuid = "8f399da3-3557-5675-b5ff-fb832c97cbdb"
+
+[[LinearAlgebra]]
+deps = ["Libdl"]
+uuid = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"

ABCD2.png

@@ -0,0 +1,7 @@
+name = "SweepContractor"
+uuid = "75a5deae-e917-4509-af32-a989148c8d5f"
+authors = ["Christopher Chubb <[email protected]>"]
+version = "0.1.0"
+
+[deps]
+LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"

Manifest.toml

@@ -0,0 +1,20 @@
+using Pkg; Pkg.add(url="https://github.com/chubbc/SweepContractor.jl")
+using SweepContractor
+
+L=16; d=2
+
+LTN = LabelledTensorNetwork{Tuple{Int,Int}}()
+for i∈1:L, j∈1:L
+    adj=Tuple{Int,Int}[];
+    i>1 && push!(adj,(i-1,j))
+    j>1 && push!(adj,(i,j-1))
+    i<L && push!(adj,(i+1,j))
+    j<L && push!(adj,(i,j+1))
+    LTN[i,j] = Tensor(adj, randn(d*ones(Int,length(adj))...), i, j)
+end
+
+println("example_2d_open")
+for χ ∈ d.^(4:9)
+    sweep = sweep_contract(LTN, χ, 2*χ; fast=true)
+    println("χ=$χ:\t", ldexp(sweep...))
+end

Project.toml

@@ -0,0 +1,21 @@
+using Pkg; Pkg.add(url="https://github.com/chubbc/SweepContractor.jl")
+using SweepContractor
+
+L=6; d=2
+
+LTN = LabelledTensorNetwork{Tuple{Int,Int}}()
+for i∈1:L, j∈1:L
+    adj=[
+        (mod1(i-1,L),mod1(j,L)),
+        (mod1(i+1,L),mod1(j,L)),
+        (mod1(i,L),mod1(j-1,L)),
+        (mod1(i,L),mod1(j+1,L))
+    ]
+    LTN[i,j] = Tensor(adj, randn(d,d,d,d), i+0.01*rand(), j+0.01*rand())
+end
+
+println("example_2d_periodic")
+for χ ∈ d.^(4:9)
+    sweep = sweep_contract(LTN,χ, 2*χ)
+    println("χ=$χ:\t",ldexp(sweep...))
+end

example_2d_open.jl

@@ -0,0 +1,27 @@
+using Pkg; Pkg.add(url="https://github.com/chubbc/SweepContractor.jl")
+using SweepContractor
+
+L=3; d=2
+
+LTN = LabelledTensorNetwork{Tuple{Int,Int,Int}}()
+for i∈1:L, j∈1:L, k∈1:L
+    adj=Tuple{Int,Int,Int}[];
+    i>1 && push!(adj,(i-1,j,k))
+    i<L && push!(adj,(i+1,j,k))
+    j>1 && push!(adj,(i,j-1,k))
+    j<L && push!(adj,(i,j+1,k))
+    k>1 && push!(adj,(i,j,k-1))
+    k<L && push!(adj,(i,j,k+1))
+    LTN[i,j,k] = Tensor(
+        adj,
+        randn(d*ones(Int,length(adj))...),
+        i+0.01*randn(),
+        j+0.01*randn()
+    )
+end
+
+println("example_3d")
+for χ∈d.^(4:9)
+    sweep=sweep_contract(LTN, χ, 2*χ)
+    println("χ=$χ:\t",ldexp(sweep...))
+end

example_2d_periodic.jl

@@ -0,0 +1,20 @@
+using Pkg; Pkg.add(url="https://github.com/chubbc/SweepContractor.jl")
+using SweepContractor
+
+LTN = LabelledTensorNetwork{Char}()
+LTN['A'] = Tensor(['D','B'], [i^2-2j for i=0:2, j=0:2], 0, 1)
+LTN['B'] = Tensor(['A','D','C'], [-3^i*j+k for i=0:2, j=0:2, k=0:2], 0, 0)
+LTN['C'] = Tensor(['B','D'], [j for i=0:2, j=0:2], 1, 0)
+LTN['D'] = Tensor(['A','B','C'], [i*j*k for i=0:2, j=0:2, k=0:2], 1, 1)
+
+brute = 0.0
+for ab=1:3, ad=1:3, bc=1:3, bd=1:3, cd=1:3
+    global brute += LTN['A'].arr[ad,ab] * LTN['B'].arr[ab,bd,bc] * LTN['C'].arr[bc,cd] *
+        LTN['D'].arr[ad,bd,cd]
+end
+
+sweep = sweep_contract(LTN, 2, 4; fast=true)
+
+println("example_ABCD")
+println("brute = $brute")
+println("sweep = $(ldexp(sweep...))")

example_3d.jl

@@ -0,0 +1,20 @@
+using Pkg; Pkg.add(url="https://github.com/chubbc/SweepContractor.jl")
+using SweepContractor
+
+n=8; d=2
+
+TN=TensorNetwork(undef,n)
+for i=1:n
+    TN[i]=Tensor(
+        setdiff(1:n,i),
+        randn(d*ones(Int,n-1)...),
+        randn(),
+        randn()
+    )
+end
+
+println("example_rand")
+for χ∈d.^(4:9)
+    sweep=sweep_contract(TN,χ,2*χ;connected=false)
+    println("χ=$χ:\t",ldexp(sweep...))
+end

example_ABCD.jl

@@ -0,0 +1,21 @@
+module SweepContractor
+    import Base: show, print, summary
+    export Tensor, TensorNetwork, LabelledTensorNetwork
+    # Define the types
+    include("tensor_network.jl")
+
+    import Base: By, Lt, Ordering, Forward, ForwardOrdering, ReverseOrdering, lt
+    include("sorted_data_structs/data_structures.jl")
+    include("sorted_data_structs/balanced_tree.jl")
+    include("sorted_data_structs/avl_tree.jl")
+    include("sorted_data_structs/sorted_dict.jl")
+    include("sorted_data_structs/priorityqueue.jl")
+    import LinearAlgebra
+    import Base: issorted, sortperm, sort!, isless, intersect
+    # Utilities to make sure networks are suitably formatted for sweep contraction
+    include("tn_utilities.jl")
+
+    export sweep_contract!, sweep_contract
+    # The sweep contraction algorithm
+    include("sweep_contract.jl")
+end