Skip to content

Commit e0d3afa

Browse files
authored
Merge pull request #66 from FluxML/kf/psizeinfer
Make psize inferable
2 parents 5954d19 + 28715b4 commit e0d3afa

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/impl/conv.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,9 @@
22
function psize(p, x)
33
nd = ndims(x)-2
44
if isa(p,Number)
5-
fill(Int(p),nd)
5+
ntuple(_->Int(p), nd)
66
elseif length(p)==nd
7-
collect(Int,p)
7+
tuple(p...)
88
else
99
throw(DimensionMismatch("psize: $p $nd"))
1010
end

0 commit comments

Comments
 (0)