Add similarterm for array

JuliaSymbolics · Oct 6, 2023 · d7c0fbe · d7c0fbe
1 parent a206b10
commit d7c0fbe
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/arrays.jl b/src/arrays.jl
@@ -92,6 +92,14 @@ function arguments(a::ArrayOp)
     arguments(a.term)
 end
 
+function similarterm(a::ArrayOp, f, args, type; metadata=nothing)
+    res = f(args...)
+    if res isa Symbolic && metadata !== nothing
+        res = SymbolicUtils.metadata(res, metadata)
+    end
+    res
+end
+
 function Base.isequal(a::ArrayOp, b::ArrayOp)
     a === b && return true
     isequal(a.shape, b.shape) &&