ton/IronpenGPU
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add validate()

Browse Source
This commit is contained in:
ton
2023-08-06 13:56:54 +07:00
parent 302f506b5b
commit 922fb1fdd9
1 changed files with 47 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/forward.jl
View File

@@ -40,6 +40,8 @@ function (kfn::kfn_1)(input::AbstractArray)
reshape(kfn.alif_zt, (size(input, 1), :, 1, size(input, 3))), dims=2)
kfn.zit .= reshape(_zit, (size(input, 1), :, size(input, 3)))
@sync begin
@async begin
# project 3D kfn zit into 4D lif zit
i1, i2, i3, i4 = size(kfn.lif_zit)
kfn.lif_zit .= reshape(kfn.zit, (i1, i2, 1, i4)) .* kfn.lif_arrayProjection4d
@@ -58,7 +60,8 @@ function (kfn::kfn_1)(input::AbstractArray)
kfn.lif_gammaPd,
kfn.lif_firingCounter,
kfn.lif_recSignal,)
end
@async begin
# project 3D kfn zit into 4D alif zit
i1, i2, i3, i4 = size(kfn.alif_zit)
kfn.alif_zit .= reshape(kfn.zit, (i1, i2, 1, i4)) .* kfn.alif_arrayProjection4d
@@ -82,6 +85,8 @@ function (kfn::kfn_1)(input::AbstractArray)
kfn.alif_avth,
kfn.alif_beta,
kfn.alif_rho,)
end
end
# reduce lif_zt4d and alif_zt4d into lif_zt, alif_zt (4d -> 1d)
kfn.lif_zt .= reduce(max, kfn.lif_zt4d, dims=(1,2))