minor fix

2023-08-15 19:43:30 +07:00
parent 1775684e38
commit 6b2bfa4628
1 changed files with 2 additions and 1 deletions
--- a/src/forward.jl
+++ b/src/forward.jl
@@ -126,7 +126,8 @@ function (kfn::kfn_1)(input::AbstractArray)
                kfn.on_recSignal,
                kfn.on_subscription,
                )
-    
+    # get on_zt4d to on_zt
+    kfn.on_zt .= reduce(max, kfn.on_zt4d, dims=(1,2))
    logit = reshape(kfn.on_zt, (size(input, 1), :))

 #     error("DEBUG -> kfn forward")