add getindex(::FiniteMPS, ::AbstractUnitRange)

QuantumKitHub · Dec 20, 2024 · 13e9ea9 · 13e9ea9
1 parent 1ed8370
commit 13e9ea9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/states/finitemps.jl b/src/states/finitemps.jl
@@ -307,6 +307,11 @@ Base.@propagate_inbounds function Base.getindex(ψ::FiniteMPS, i::Int)
     end
 end
 
+Base.@propagate_inbounds function Base.getindex(ψ::FiniteMPS, I::AbstractUnitRange)
+    return Base.getindex.(Ref(ψ), I)
+end
+
+
 function Base.convert(::Type{TensorMap}, ψ::FiniteMPS)
     T = foldl(ψ.AR[2:end]; init=first(ψ.AC)) do x, y
         return _transpose_front(x * _transpose_tail(y))