From ff9909fd8dc8a899dc5855008afd59378977885f Mon Sep 17 00:00:00 2001 From: ton Date: Mon, 24 Jul 2023 14:25:57 +0700 Subject: [PATCH] kfn forward() --- src/forward.jl | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/forward.jl b/src/forward.jl index 162f95b..daa76ed 100644 --- a/src/forward.jl +++ b/src/forward.jl @@ -73,7 +73,7 @@ function (kfn::kfn_1)(input::AbstractArray) reshape(kfn.alif_zt1, (d1, :, 1, d3)), dims=2) kfn.zit .= reshape(_zit, (d1, :, d3)) - #WORKING read out + # read out onForward( kfn.zit, kfn.on_zit, kfn.on_wRec, @@ -231,7 +231,7 @@ function onForward(kfn_zit, zit .= reshape(kfn_zit, (d1, d2, 1, d4)) .* ones(size(wRec)...) # project zit into zit for j in 1:d4, i in 1:d3 # compute along neurons axis of every batch - if view(refractoryCounter, :, :, i, j)[1] > 0 # refractory period is active + if view(refractoryCounter, :, :, i, j)[1] > 0 # neuron is inactive (in refractory period) view(refractoryCounter, :, :, i, j)[1] -= 1 view(zt1, :, :, i, j)[1] = 0 view(vt1, :, :, i, j)[1] = @@ -239,7 +239,7 @@ function onForward(kfn_zit, view(phi, :, :, i, j)[1] = 0.0 view(epsilonRec, :, :, i, j) .= view(alpha, :, :, i, j)[1] .* view(epsilonRec, :, :, i, j) - else # refractory period is inactive + else # neuron is active view(vt1, :, :, i, j)[1] = (view(alpha, :, :, i, j)[1] * view(vt0,:, :, i, j)[1]) + sum(view(zit, :, :, i, j) .* view(wRec, :, :, i, j))