more dims

Author: lkdvos

src/spaces/homspace.jl

@@ -127,6 +127,8 @@ function dim(W::HomSpace)
     return d
 end
 
+dims(W::HomSpace) = (dims(codomain(W))..., dims(domain(W))...)
+
 """
     fusiontrees(W::HomSpace)
 

src/spaces/productspace.jl

@@ -48,11 +48,14 @@ end
 
 # Corresponding methods
 #-----------------------
-"""
+@doc """
     dims(::ProductSpace{S, N}) -> Dims{N} = NTuple{N, Int}
+    dims(V::HomSpace) -> Dims{length(V)}
+    dims(t::AbstractTensorMap) -> Dims{numind(t)}
+
+Return the dimensions of the spaces in the tensor product space(s) as a tuple of integers.
+""" dims
 
-Return the dimensions of the spaces in the tensor product space as a tuple of integers.
-"""
 dims(P::ProductSpace) = map(dim, P.spaces)
 dim(P::ProductSpace, n::Int) = dim(P.spaces[n])
 dim(P::ProductSpace) = prod(dims(P))

src/tensors/abstracttensor.jl

@@ -195,6 +195,8 @@ symmetry. This is also the dimension of the `HomSpace` on which the `TensorMap`
 """
 dim(t::AbstractTensorMap) = fusionblockstructure(t).totaldim
 
+dims(t::AbstractTensorMap) = dims(space(t))
+
 """
     blocksectors(t::AbstractTensorMap)
 
@@ -461,8 +463,7 @@ end
     Base.getindex(t::AbstractTensorMap)
     t[]
 
-Return a view into the data of `t` as a `StridedViews.StridedView` of size
-`(dims(codomain(t))..., dims(domain(t))...)`.
+Return a view into the data of `t` as a `StridedViews.StridedView` of size `dims(t)`.
 """
 @inline function Base.getindex(t::AbstractTensorMap)
     return t[trivial_fusiontree(t)...]
@@ -616,7 +617,7 @@ function Base.convert(::Type{Array}, t::AbstractTensorMap)
         dom = domain(t)
         T = sectorscalartype(I) <: Complex ? complex(scalartype(t)) :
             sectorscalartype(I) <: Integer ? scalartype(t) : float(scalartype(t))
-        A = zeros(T, dims(cod)..., dims(dom)...)
+        A = zeros(T, dims(t)...)
         for (f₁, f₂) in fusiontrees(t)
             F = convert(Array, (f₁, f₂))
             Aslice = StridedView(A)[axes(cod, f₁.uncoupled)..., axes(dom, f₂.uncoupled)...]

src/tensors/tensor.jl

@@ -333,7 +333,7 @@ function TensorMap(
     # dimension check
     codom = codomain(V)
     dom = domain(V)
-    arraysize = (dims(codom)..., dims(dom)...)
+    arraysize = dims(V)
     matsize = (dim(codom), dim(dom))
 
     if !(size(data) == arraysize || size(data) == matsize)
@@ -487,7 +487,7 @@ end
     @boundscheck begin
         sectortype(t) == Trivial || throw(SectorMismatch())
     end
-    return sreshape(StridedView(t.data), (dims(codomain(t))..., dims(domain(t))...))
+    return sreshape(StridedView(t.data), dims(t))
 end
 
 # Show