fix cat

Author: CarloLucibello

src/upsample.jl

@@ -120,18 +120,12 @@ function ∇bilinear_upsample(Δ::AbstractArray{<:Number, 4}, k::NTuple{2,Int})
 
     kern1 = get_downsamplekernel(Δ, k[1])
     kern2 = get_downsamplekernel(Δ, k[2])
-    kern = kern1 .* kern2'
+    kern = kern1 * kern2'
 
     pad = (floor(Int, k[1]//2), floor(Int, k[2]//2))
     stride = k
 
-    weight = similar(Δ, eltype(Δ), (size(kern)..., n_chan, n_chan))
-    weight .= 0
-    for i in 1:n_chan
-        weight[:,:,i,i] .= kern
-    end
-    # weight = cat(fill(kern, n_chan), dims=(3,4)) # produces Array{Any}, revisit in the future
-    
+    weight = cat(fill(kern, n_chan)..., dims=(3,4)) 
     dx = conv(Δ, weight, pad=pad, stride=stride)
 
     # Still have to fix edge effects due to zero-padding of convolution,