padding

KnetML · Oct 27, 2023 · 638b284 · 638b284
1 parent c100764
commit 638b284
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -15,6 +15,7 @@ data/mnist
 data/fashion_mnist
 data/pfam
 test/logs
+test/save_lenet.jld2
 codecov.token
 logs
 .vscode

diff --git a/src/layers.jl b/src/layers.jl
@@ -479,7 +479,7 @@ function (l::Pad)(x)
     core = Tuple( (1+pdims[i]) : (size(x,i)+pdims[i]) for i in 1:nd)
 
     padded = fill!(similar(x, siz...), value)
-    padded[core...] = x
+    padded[core...] .= x
     return padded
 end