Add HyperGroup transformation

src/Tensor.jl

@@ -89,6 +89,7 @@ Base.replace(t::Tensor, old_new::Pair{Symbol,Symbol}...) = Tensor(parent(t), rep
 
 Base.parent(t::Tensor) = t.data
 parenttype(::Type{Tensor{T,N,A}}) where {T,N,A} = A
+parenttype(::T) where {T<:Tensor} = parenttype(T)
 
 dim(::Tensor, i::Number) = i
 dim(t::Tensor, i::Symbol) = first(findall(==(i), inds(t)))

src/Transformations.jl

@@ -41,6 +41,8 @@ end
 
 Convert hyperindices to COPY-tensors, represented by `DeltaArray`s.
 This transformation is always used by default when visualizing a `TensorNetwork` with `plot`.
+
+See also: [`HyperGroup`](@ref).
 """
 struct HyperFlatten <: Transformation end
 
@@ -72,6 +74,35 @@ function transform!(tn::TensorNetwork, ::HyperFlatten)
     end
 end
 
+"""
+    HyperGroup <: Transformation
+
+Convert COPY-tensors, represented by `DeltaArray`s, to hyperindices.
+
+See also: [`HyperFlatten`](@ref).
+"""
+struct HyperGroup <: Transformation end
+
+function transform!(tn::TensorNetwork, ::HyperGroup)
+    targets = Iterators.filter(x -> parenttype(x) isa DeltaArray, tensors(tn))
+    for tensor in targets
+        # remove COPY tensor
+        delete!(tn, target)
+
+        # insert hyperindex
+        hyperindex = uuid4()
+
+        # insert weights vector
+        if !all(isone, delta(parent(tensor)))
+            push!(tn, Tensor(delta(parent(tensor)), [hyperindex]))
+        end
+
+        for flatindex in inds(tensor)
+            replace!(tn, flatindex => hyperindex)
+        end
+    end
+end
+
 """
     DiagonalReduction <: Transformation