the carefulest!

Author: lkdvos

src/tensors/abstracttensor.jl

@@ -82,6 +82,8 @@ similarstoragetype(::Type{TT}, ::Type{T}) where {TT <: AbstractTensorMap, T <: N
 
 # implement on arrays
 similarstoragetype(::Type{A}) where {A <: DenseVector{<:Number}} = A
+Base.@assume_effects :foldable similarstoragetype(::Type{A}) where {A <: AbstractArray{<:Number}} =
+    Core.Compiler.return_type(similar, Tuple{A, Int})
 Base.@assume_effects :foldable similarstoragetype(::Type{A}, ::Type{T}) where {A <: AbstractArray, T <: Number} =
     Core.Compiler.return_type(similar, Tuple{A, Type{T}, Int})
 
@@ -544,17 +546,17 @@ end
 
 # 3 arguments
 Base.similar(t::AbstractTensorMap, codomain::TensorSpace, domain::TensorSpace) =
-    similar(t, similarstoragetype(t), codomain ← domain)
+    similar(t, similarstoragetype(t, scalartype(t)), codomain ← domain)
 Base.similar(t::AbstractTensorMap, ::Type{T}, codomain::TensorSpace) where {T} =
     similar(t, T, codomain ← one(codomain))
 
 # 2 arguments
 Base.similar(t::AbstractTensorMap, codomain::TensorSpace) =
-    similar(t, similarstoragetype(t), codomain ← one(codomain))
-Base.similar(t::AbstractTensorMap, V::TensorMapSpace) = similar(t, similarstoragetype(t), V)
+    similar(t, codomain ← one(codomain))
+Base.similar(t::AbstractTensorMap, V::TensorMapSpace) = similar(t, scalartype(t), V)
 Base.similar(t::AbstractTensorMap, ::Type{T}) where {T} = similar(t, T, space(t))
 # 1 argument
-Base.similar(t::AbstractTensorMap) = similar(t, similarstoragetype(t), space(t))
+Base.similar(t::AbstractTensorMap) = similar(t, scalartype(t), space(t))
 
 # generic implementation for AbstractTensorMap -> returns `TensorMap`
 function Base.similar(t::AbstractTensorMap, ::Type{TorA}, V::TensorMapSpace) where {TorA}
@@ -565,7 +567,7 @@ end
 
 # implementation in type-domain
 function Base.similar(::Type{TT}, V::TensorMapSpace) where {TT <: AbstractTensorMap}
-    TT′ = tensormaptype(spacetype(V), numout(V), numin(V), similarstoragetype(TT))
+    TT′ = tensormaptype(spacetype(V), numout(V), numin(V), similarstoragetype(TT, scalartype(TT)))
     return TT′(undef, V)
 end
 Base.similar(::Type{TT}, cod::TensorSpace, dom::TensorSpace) where {TT <: AbstractTensorMap} =