added version of derivative that takes a scalar argument and returns …

…a scalar output.
brianguenter · Nov 30, 2023 · 5642c1a · 5642c1a
1 parent ce661db
commit 5642c1a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/Jacobian.jl b/src/Jacobian.jl
@@ -385,6 +385,12 @@ function derivative(A::AbstractArray{<:Node}, variables::T...) where {T<:Node}
 end
 export derivative
 
+"""Convenience `derivative` for scalar functions. Takes a scalar input and returns a scalar output"""
+function derivative(A::Node, variables::T...) where {T<:Node}
+    derivative([A], variables...)[1]
+end
+
+
 """
     _derivative(A::AbstractArray{<:Node}, variable::Node)