Format

Author: mtfishman

Project.toml

@@ -1,7 +1,7 @@
 name = "ITensorBase"
 uuid = "4795dd04-0d67-49bb-8f44-b89c448a1dc7"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.1.3"
+version = "0.1.4"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"

src/ITensorBase.jl

@@ -54,6 +54,7 @@ addtags(n::IndexName, ts) = settags(n, tags(n) ∪ tagset(ts))
 
 setprime(n::IndexName, plev) = @set n.plev = plev
 prime(n::IndexName) = setprime(n, plev(n) + 1)
+noprime(n::IndexName) = setprime(n, 0)
 
 function Base.show(io::IO, i::IndexName)
   idstr = "id=$(id(i) % 1000)"
@@ -97,6 +98,7 @@ plev(i::Index) = plev(name(i))
 addtags(i::Index, tags) = setname(i, addtags(name(i), tags))
 prime(i::Index) = setname(i, prime(name(i)))
 Base.adjoint(i::Index) = prime(i)
+noprime(i::Index) = setname(i, noprime(name(i)))
 
 # Interface
 # TODO: Overload `Base.parent` instead.
@@ -253,6 +255,7 @@ end
 # TODO: Use `replaceinds`/`mapinds`, based on
 # `replacenameddimsindices`/`mapnameddimsindices`.
 prime(a::AbstractITensor) = setinds(a, prime.(inds(a)))
+noprime(a::AbstractITensor) = setinds(a, noprime.(inds(a)))
 
 include("quirks.jl")
 

src/quirks.jl

@@ -1,14 +1,18 @@
 # TODO: Define this properly.
 dag(i::Index) = i
+dag(a::ITensor) = a
 # TODO: Deprecate.
 dim(i::Index) = dename(length(i))
+dim(a::AbstractITensor) = unname(length(a))
 # TODO: Define this properly.
 hasqns(i::Index) = false
 # TODO: Deprecate.
 itensor(parent::AbstractArray, nameddimsindices) = ITensor(parent, nameddimsindices)
 function itensor(parent::AbstractArray, i1::Index, i_rest::Index...)
   return ITensor(parent, (i1, i_rest...))
 end
+# TODO: Deprecate.
+order(a::AbstractArray) = ndims(a)
 
 # This seems to be needed to get broadcasting working.
 # TODO: Investigate this and see if we can get rid of it.
@@ -23,9 +27,25 @@ function onehot(iv::Pair{<:Index,<:Int})
   return a
 end
 
-using LinearAlgebra: svd
+using LinearAlgebra: qr, svd
 # TODO: Define this in `MatrixAlgebra.jl`/`TensorAlgebra.jl`.
-function factorize(a::AbstractITensor, args...; kwargs...)
-  U, S, V = svd(a, args...; kwargs...)
-  return U, S * V
+function factorize(a::AbstractITensor, args...; tags=TagSet(), maxdim=nothing)
+  if isnothing(maxdim)
+    # `tags` are ignored right now.
+    Q, R = qr(a, args...)
+    return Q, R
+  else
+    error("Truncation in `factorize` not implemented yet.")
+    U, S, V = svd(a, args...; kwargs...)
+    return U, S * V
+  end
 end
+
+# TODO: Used in `ITensorMPS.jl`, decide where or if to define it.
+# Ideally this would just be a zero-dimensional `ITensor` wrapping
+# a special type, like `Zeros{UnspecifiedZero()}()`.
+struct OneITensor <: AbstractITensor end
+Base.size(::OneITensor) = ()
+Base.:*(::OneITensor, ::OneITensor) = OneITensor()
+Base.:*(::OneITensor, a::ITensor) = a
+Base.:*(a::ITensor, ::OneITensor) = a