failed to run `@example` block in src/symbolic_neural_networks.md:86-96 ```@example snn import Random # hide Random.seed!(123) # hide nn_cpu = NeuralNetwork(c, CPU()) o = Optimizer(AdamOptimizer(), nn_cpu) n_epochs = 1000 batch = Batch(10) o(nn_cpu, dl, batch, n_epochs, pb.loss, pb; show_progress = false); # hide @time o(nn_cpu, dl, batch, n_epochs, pb.loss, pb; show_progress = false); nothing # hide ``` exception = DimensionMismatch: new dimensions (3, 1, 10) must be consistent with array size 60 Stacktrace: [1] (::Base.var"#throw_dmrsa#365")(dims::Tuple{Int64, Int64, Int64}, len::Int64) @ Base ./reshapedarray.jl:41 [2] reshape @ ./reshapedarray.jl:45 [inlined] [3] reshape(::Matrix{Float64}, ::Int64, ::Int64, ::Int64) @ Base ./reshapedarray.jl:127 [4] (::SymbolicNeuralNetworks.var"#gen_fun_returned#19"{RuntimeGeneratedFunctions.RuntimeGeneratedFunction{(:sinput, :soutput, :ps, :k), SymbolicNeuralNetworks.var"#_RGF_ModTag", SymbolicNeuralNetworks.var"#_RGF_ModTag", (0xeb6db357, 0xab1500c0, 0xe492125b, 0x1b9007b0, 0x3ee7d6f5), Expr}})(x::Array{Float64, 3}, y::Array{Float64, 3}, ps::AbstractNeuralNetworks.NeuralNetworkParameters{(:L1, :L2, :L3), Tuple{@NamedTuple{W::Matrix{Float64}, b::Vector{Float64}}, @NamedTuple{W::Matrix{Float64}, b::Vector{Float64}}, @NamedTuple{W::Matrix{Float64}, b::Vector{Float64}}}}) @ SymbolicNeuralNetworks ~/work/SymbolicNeuralNetworks.jl/SymbolicNeuralNetworks.jl/src/utils/build_function2.jl:30 [5] macro expansion @ ~/work/SymbolicNeuralNetworks.jl/SymbolicNeuralNetworks.jl/src/utils/build_function_arrays.jl:207 [inlined] [6] apply_element_wise @ ~/work/SymbolicNeuralNetworks.jl/SymbolicNeuralNetworks.jl/src/utils/build_function_arrays.jl:207 [inlined] [7] _pbs_executable @ ~/work/SymbolicNeuralNetworks.jl/SymbolicNeuralNetworks.jl/src/utils/build_function_arrays.jl:77 [inlined] [8] ___pbs_executable @ ~/work/SymbolicNeuralNetworks.jl/SymbolicNeuralNetworks.jl/src/utils/build_function_arrays.jl:80 [inlined] [9] macro expansion @ ~/work/SymbolicNeuralNetworks.jl/SymbolicNeuralNetworks.jl/src/utils/build_function_arrays.jl:248 [inlined] [10] apply_element_wise(ps::AbstractNeuralNetworks.NeuralNetworkParameters{(:L1, :L2, :L3), Tuple{SymbolicNeuralNetworks.var"#___pbs_executable#27"{SymbolicNeuralNetworks.var"#_pbs_executable#25", @NamedTuple{W::SymbolicNeuralNetworks.var"#gen_fun_returned#19"{RuntimeGeneratedFunctions.RuntimeGeneratedFunction{(:sinput, :soutput, :ps, :k), SymbolicNeuralNetworks.var"#_RGF_ModTag", SymbolicNeuralNetworks.var"#_RGF_ModTag", (0xeb6db357, 0xab1500c0, 0xe492125b, 0x1b9007b0, 0x3ee7d6f5), Expr}}, b::SymbolicNeuralNetworks.var"#gen_fun_returned#19"{RuntimeGeneratedFunctions.RuntimeGeneratedFunction{(:sinput, :soutput, :ps, :k), SymbolicNeuralNetworks.var"#_RGF_ModTag", SymbolicNeuralNetworks.var"#_RGF_ModTag", (0xf9456a58, 0x1cb7f49a, 0x7d278da6, 0x34bea467, 0x8b2b3b26), Expr}}}}, SymbolicNeuralNetworks.var"#___pbs_executable#27"{SymbolicNeuralNetworks.var"#_pbs_executable#25", @NamedTuple{W::SymbolicNeuralNetworks.var"#gen_fun_returned#19"{RuntimeGeneratedFunctions.RuntimeGeneratedFunction{(:sinput, :soutput, :ps, :k), SymbolicNeuralNetworks.var"#_RGF_ModTag", SymbolicNeuralNetworks.var"#_RGF_ModTag", (0x0ad92299, 0xd3bc2b65, 0x8c83ac67, 0x71f04265, 0x87906263), Expr}}, b::SymbolicNeuralNetworks.var"#gen_fun_returned#19"{RuntimeGeneratedFunctions.RuntimeGeneratedFunction{(:sinput, :soutput, :ps, :k), SymbolicNeuralNetworks.var"#_RGF_ModTag", SymbolicNeuralNetworks.var"#_RGF_ModTag", (0x6b4c65bb, 0x865e68d0, 0x3409eed3, 0x23d1325c, 0x0bec56ad), Expr}}}}, SymbolicNeuralNetworks.var"#___pbs_executable#27"{SymbolicNeuralNetworks.var"#_pbs_executable#25", @NamedTuple{W::SymbolicNeuralNetworks.var"#gen_fun_returned#19"{RuntimeGeneratedFunctions.RuntimeGeneratedFunction{(:sinput, :soutput, :ps, :k), SymbolicNeuralNetworks.var"#_RGF_ModTag", SymbolicNeuralNetworks.var"#_RGF_ModTag", (0x1c43db35, 0x7da1a4a8, 0x281a9715, 0xcf9f272b, 0x8c06