Split out N=1 in Khatri-rao

dahong67 · Jan 16, 2024 · 48d965e · 48d965e
1 parent 6d33f3b
commit 48d965e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/gcp-opt.jl b/src/gcp-opt.jl
@@ -219,6 +219,9 @@ end
     of the matrices `A1`, `A2`, etc.
 """
 function khatrirao(A::Vararg{T,N}) where {T<:AbstractMatrix,N}
+    if N == 1
+        return A[1]
+    end
     r = size(A[1],2)
     # @boundscheck all(==(r),size.(A,2)) || throw(DimensionMismatch())
     R = ntuple(Val(N)) do k