Make sure result is an Array

dahong67 · Aug 20, 2024 · 7fda495 · 7fda495
1 parent 8c31d59
commit 7fda495
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/GCPDecompositions.jl b/src/GCPDecompositions.jl
@@ -6,7 +6,7 @@ module GCPDecompositions
 # Imports
 import Base: ndims, size, show, summary
 import Base: getindex
-import Base: Array
+import Base: AbstractArray, Array
 import LinearAlgebra: norm
 using IntervalSets: Interval
 using Random: default_rng

diff --git a/src/cpd.jl b/src/cpd.jl
@@ -87,7 +87,8 @@ function getindex(M::CPD{T,N}, I::Vararg{Int,N}) where {T,N}
 end
 getindex(M::CPD{T,N}, I::CartesianIndex{N}) where {T,N} = getindex(M, Tuple(I)...)
 
-Array(A::CPD) = reshape(TensorKernels.khatrirao(reverse(A.U)...) * A.λ, size(A))
+AbstractArray(A::CPD) = reshape(TensorKernels.khatrirao(reverse(A.U)...) * A.λ, size(A))
+Array(A::CPD) = Array(AbstractArray(A))
 
 norm(M::CPD, p::Real = 2) =
     p == 2 ? norm2(M) : norm((M[I] for I in CartesianIndices(size(M))), p)