use the more general outputsize

JuliaSymbolics · Feb 27, 2024 · d3ebacd · d3ebacd
1 parent 5dfa5f5
commit d3ebacd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ext/SymbolicsLuxCoreExt.jl b/ext/SymbolicsLuxCoreExt.jl
@@ -4,7 +4,7 @@ using LuxCore, Symbolics
 
 @register_array_symbolic LuxCore.stateless_apply(
     model::LuxCore.AbstractExplicitLayer, x::AbstractArray, ps::Union{NamedTuple, <:AbstractVector}) begin
-    size = LuxCore.outputsize(model)
+    size = LuxCore.outputsize(model, x, LuxCore._default_rng())
     eltype = Real
 end