Merge pull request #99 from vpuri3/multidim

function operator multidim

Author: ChrisRackauckas

src/func.jl

@@ -66,6 +66,28 @@ mutable struct FunctionOperator{iip,oop,T<:Number,F,Fa,Fi,Fai,Tr,P,Tt,C} <: Abst
     end
 end
 
+function FunctionOperator(op,
+                          input::AbstractArray{<:Any,D},
+                          output::AbstractArray{<:Any,D};
+                          kwargs...) where{D}
+    D ≤ 2 && @error "FunctionOperator not defined for $(typeof(input)), $(typeof(output))."
+
+    NK = length(input)
+    MK = length(output)
+
+    M = size(output, 1)
+    N = size(input, 1)
+
+    K  = NK ÷ N
+
+    @assert MK == M * K
+
+    input  = reshape(input,  (N, K))
+    output = reshape(output, (M, K))
+
+    FunctionOperator(op, input, output; kwargs...)
+end
+
 function FunctionOperator(op,
                           input::AbstractVecOrMat,
                           output::AbstractVecOrMat;

src/interface.jl

@@ -51,6 +51,8 @@ function cache_operator(L::AbstractSciMLOperator, u::AbstractVecOrMat)
 end
 
 function cache_operator(L::AbstractSciMLOperator, u::AbstractArray)
+    u isa AbstractVecOrMat && @error "cache_operator not defined for $(typeof(L)), $(typeof(u))."
+
     n = size(L, 2)
     s = size(u)
     k = prod(s[2:end])