Add missing similar

lkdvos · Nov 15, 2024 · a8f7ff0 · a8f7ff0
1 parent 3489333
commit a8f7ff0
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/tensors/abstractblocktensor/abstractarray.jl b/src/tensors/abstractblocktensor/abstractarray.jl
@@ -209,6 +209,15 @@ function similar_tensormaptype(
         return AbstractTensorMap{scalartype(T),S,numout(P),numin(P)}
     end
 end
+function similar_tensormaptype(
+    t::AbstractBlockTensorMap, M::Type{<:AbstractVector}, P::TensorMapSumSpace{S}
+) where {S}
+    if isconcretetype(eltype(t))
+        return tensormaptype(S, numout(P), numin(P), M)
+    else
+        return AbstractTensorMap{scalartype(M),S,numout(P),numin(P)}
+    end
+end
 function similar_tensormaptype(
     t::AbstractBlockTensorMap, T::Type{<:Number}, P::TensorMapSumSpace{S}
 ) where {S}